Pushing back the limits on numerical computing.
15
सार्वजनिक रिपोजिटरी
39,172
कुल सितारे
968
अनुयायी
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.
इस रिपोजिटरी के लिए कोई विवरण प्रदान नहीं किया गया।
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
इस रिपोजिटरी के लिए कोई विवरण प्रदान नहीं किया गया।
इस रिपोजिटरी के लिए कोई विवरण प्रदान नहीं किया गया।
इस रिपोजिटरी के लिए कोई विवरण प्रदान नहीं किया गया।
इस रिपोजिटरी के लिए कोई विवरण प्रदान नहीं किया गया।
इस रिपोजिटरी के लिए कोई विवरण प्रदान नहीं किया गया।
RepoGuard के साथ jax-ml की निगरानी करें और जैसे ही एक नया सार्वजनिक रिपोजिटरी बनता है, सूचित हों।
इस खाते की निगरानी करें