Members
-
jax ★ PINNED
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python ★ 36k 1h agoExplain → -
jax-llm-examples ★ PINNED
Minimal yet performant LLM examples in pure JAX
Python ★ 267 6d agoExplain → -
jax-triton ★ PINNED
jax-triton contains integrations between JAX and OpenAI Triton
Python ★ 465 5h agoExplain → -
scaling-book ★ PINNED
Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
HTML ★ 1.3k 2d agoExplain → -
ml_dtypes ★ PINNED
A stand-alone implementation of several NumPy dtype extensions used in machine learning.
C++ ★ 350 10d agoExplain → -
oryx
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
Python ★ 319 2d agoExplain → -
jax-ai-stack
No description.
Python ★ 303 11d agoExplain → -
bayeux
State of the art inference for your bayesian models.
Python ★ 245 5mo agoExplain → -
bonsai
Minimal, lightweight JAX implementations of popular models.
Python ★ 239 1mo agoExplain → -
coix
Inference Combinators in JAX
Jupyter Notebook ★ 53 1y agoExplain → -
jax-tpu-embedding
No description.
Python ★ 34 8h agoExplain → -
australis ▣
No description.
C++ ★ 28 3y agoExplain → -
jax-blog
No description.
★ 6 6mo agoExplain → -
.allstar ▣
No description.
★ 0 3y agoExplain → -
.github ▣
No description.
★ 0 3y agoExplain →
No repos match these filters.