Pinned Loading

1. jax jax Public

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

   Python 34.2k 3.3k

2. jax-llm-examples jax-llm-examples Public

   Minimal yet performant LLM examples in pure JAX

   Python 206 27

3. jax-triton jax-triton Public

   jax-triton contains integrations between JAX and OpenAI Triton

   Python 436 53

4. scaling-book scaling-book Public

   Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs

   HTML 719 104

5. ml_dtypes ml_dtypes Public

   A stand-alone implementation of several NumPy dtype extensions used in machine learning.

   C++ 317 51