Pinned Loading

1. jax jax Public

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

   Python 33.3k 3.1k

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

   Minimal yet performant LLM examples in pure JAX

   Python 150 19

3. jax-triton jax-triton Public

   jax-triton contains integrations between JAX and OpenAI Triton

   Python 415 50

4. scaling-book scaling-book Public

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

   HTML 562 85

5. ml_dtypes ml_dtypes Public

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

   C++ 292 45