Pushing back the limits on numerical computing.
15
Publiczne repozytoria
39 172
Łączna liczba gwiazdek
968
Obserwujący
Najpopularniejsze języki
Publiczne repozytoria
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
scaling-book
Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
jax-triton
jax-triton contains integrations between JAX and OpenAI Triton
ml_dtypes
A stand-alone implementation of several NumPy dtype extensions used in machine learning.
oryx
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
jax-ai-stack
Brak opisu dla tego repozytorium.
jax-llm-examples
Minimal yet performant LLM examples in pure JAX
bayeux
State of the art inference for your bayesian models.
bonsai
Minimal, lightweight JAX implementations of popular models.
coix
Inference Combinators in JAX
jax-tpu-embedding
Brak opisu dla tego repozytorium.
australis
Brak opisu dla tego repozytorium.
jax-blog
Brak opisu dla tego repozytorium.
.allstar
Brak opisu dla tego repozytorium.
.github
Brak opisu dla tego repozytorium.
Czy to ujawnienie jest zamierzone?
Monitoruj jax-ml z RepoGuard i otrzymuj powiadomienia w momencie, gdy pojawi się nowe publiczne repozytorium.
Monitoruj to konto