• 3
• 0
• 0

kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

• 2
• 0
• 0

reverb

Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research

• 3
• 0
• 0

kfac_jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

• 3
• 0
• 0

kapitan

Generic templated configuration management for Kubernetes, Terraform and other things

• 3
• 0
• 0

mujoco

Multi-Joint dynamics with Contact. A general purpose physics simulator.

• 2
• 0
• 0

optax

Optax is a gradient processing and optimization library for JAX.

• 2
• 0
• 0

open_spiel

OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.

• 3
• 0
• 0

jax_privacy

Algorithms for Privacy-Preserving Machine Learning in JAX

• 3
• 0
• 0

xmanager

A platform for managing machine learning experiments

• 3
• 0
• 0

meltingpot

A suite of test scenarios for multi-agent reinforcement learning.

• 2
• 0
• 1

acme

A library of reinforcement learning components and agents

• 3
• 0
• 0

tf2jax

• 3
• 0
• 0

envlogger

A tool for recording RL trajectories.

• 3
• 0
• 0

dm_pix

PIX is an image processing library in JAX, for JAX.

• 2
• 0
• 0

dm-haiku

JAX-based neural network library

• 3
• 0
• 0

dm_nevis

NEVIS'22: Benchmarking the next generation of never-ending learners

• 3
• 0
• 0

s6

• 3
• 0
• 0

emergent_in_context_learning

• 3
• 0
• 0

physical_concepts

• 3
• 0
• 0

conformal_training

This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classifiers".