Popular repositories Loading

1. grain grain Public

   Forked from google/grain

   Python

2. JAX-Toolbox JAX-Toolbox Public

   Forked from NVIDIA/JAX-Toolbox

   JAX-Toolbox

   Python

3. xla xla Public

   Forked from openxla/xla

   A machine learning compiler for GPUs, CPUs, and ML accelerators

   C++

4. airio airio Public

   Forked from google/airio

   Python

5. jax jax Public

   Forked from jax-ml/jax

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

   Python

6. orbax orbax Public

   Forked from google/orbax

   Orbax provides common checkpointing and persistence utilities for JAX users

   Python