• 2
• 0
• 0

distrax

• 3
• 0
• 0

jax_privacy

Algorithms for Privacy-Preserving Machine Learning in JAX

• 2
• 0
• 0

tree

tree is a library for working with nested data structures

• 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

tf2jax

• 3
• 0
• 0

dm_pix

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

• 2
• 0
• 0

sonnet

TensorFlow-based neural network library

• 3
• 1
• 0

tapnet

Tracking Any Point (TAP)

• 2
• 0
• 0

reverb

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

• 2
• 0
• 0

dm-haiku

JAX-based neural network library

• 3
• 1
• 0

ithaca

Restoring and attributing ancient texts using deep neural networks

• 3
• 0
• 0

xmanager

A platform for managing machine learning experiments

• 2
• 0
• 1

acme

A library of reinforcement learning components and agents

• 3
• 0
• 0

kfac-jax

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

• 3
• 0
• 0

meltingpot

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

• 3
• 0
• 0

kfac_jax

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

• 3
• 0
• 0

dks

Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models ...