Popular repositories Loading

1. neuron_inverter_benchmark neuron_inverter_benchmark Public

   Forked from balewski/neuron_inverter_benchmark

   multivariate regression of time series with CNN

   Python

2. xla-folk xla-folk Public

   Forked from openxla/xla

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

   C++

3. JAX-Toolbox JAX-Toolbox Public

   Forked from NVIDIA/JAX-Toolbox

   JAX-Toolbox

   Python

4. jane-jax-folk jane-jax-folk Public

   Forked from jax-ml/jax

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

   Python

5. jane-maxtext jane-maxtext Public

   Forked from AI-Hypercomputer/maxtext

   A simple, performant and scalable Jax LLM!

   Python

6. optax-jane optax-jane Public

   Forked from google-deepmind/optax

   Optax is a gradient processing and optimization library for JAX.

   Python