Pushing back the limits on numerical computing.
15
Publieke repositories
39.172
Totaal aantal sterren
968
Volgers
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
jax-triton contains integrations between JAX and OpenAI Triton
A stand-alone implementation of several NumPy dtype extensions used in machine learning.
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
Geen beschrijving gegeven voor deze repository.
Minimal yet performant LLM examples in pure JAX
State of the art inference for your bayesian models.
Minimal, lightweight JAX implementations of popular models.
Inference Combinators in JAX
Geen beschrijving gegeven voor deze repository.
Geen beschrijving gegeven voor deze repository.
Geen beschrijving gegeven voor deze repository.
Geen beschrijving gegeven voor deze repository.
Geen beschrijving gegeven voor deze repository.
Monitor jax-ml met RepoGuard en krijg een waarschuwing op het moment dat er een nieuwe publieke repository verschijnt.
Monitor dit account