Pinned 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. maxtext maxtext Public

   Forked from AI-Hypercomputer/maxtext

   A simple, performant and scalable Jax LLM!

   Python

3. xla xla Public

   Forked from openxla/xla

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

   C++