Popular repositories Loading

1. jax jax Public

   Forked from jax-ml/jax

   Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

   Python

2. jax-ai-stack jax-ai-stack Public

   Forked from jax-ml/jax-ai-stack

   Python

3. flax flax Public

   Forked from google/flax

   Flax is a neural network library for JAX that is designed for flexibility.

   Jupyter Notebook

4. Build-and-Install Build-and-Install Public

   Jupyter Notebook

5. mctx mctx Public

   Forked from google-deepmind/mctx

   Monte Carlo tree search in JAX

   Python

6. chex chex Public

   Forked from google-deepmind/chex

   Python