Pushing back the limits on numerical computing.
15
Kamu depoları
39.172
Toplam yıldız
968
Takipçiler
En popüler diller
Kamu depoları
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
Bu depo için açıklama sağlanmadı.
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
Bu depo için açıklama sağlanmadı.
australis
Bu depo için açıklama sağlanmadı.
jax-blog
Bu depo için açıklama sağlanmadı.
.allstar
Bu depo için açıklama sağlanmadı.
.github
Bu depo için açıklama sağlanmadı.
Bu ifşa amaçlı mı?
jax-ml ile RepoGuard'ı izleyin ve yeni bir kamu deposu belirdiğinde anında uyarı alın.
Bu hesabı izleyin