Pinned Loading

1. jax jax Public

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

   Python 36k 3.7k

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

   Minimal yet performant LLM examples in pure JAX

   Python 267 36

3. jax-triton jax-triton Public

   jax-triton contains integrations between JAX and OpenAI Triton

   Python 466 57

4. scaling-book scaling-book Public

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

   HTML 1.3k 181

5. ml_dtypes ml_dtypes Public

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

   C++ 351 60