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. flax flax Public

   Forked from google/flax

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

   Jupyter Notebook

3. jax-ai-stack jax-ai-stack Public

   Forked from jax-ml/jax-ai-stack

   Python

4. keras-io keras-io Public

   Forked from keras-team/keras-io

   Keras documentation, hosted live at keras.io

   Jupyter Notebook

5. orbax orbax Public

   Forked from google/orbax

   Orbax provides common checkpointing and persistence utilities for JAX users

   Python

6. optax optax Public

   Forked from google-deepmind/optax

   Optax is a gradient processing and optimization library for JAX.

   Python