代码库
Oryx is a library for probabilistic programming and deep learning built on top of Jax.
Python
jaxprobabilistic-programmingpython
Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
SCSS
jaxllm-inferencellmsrooflinetpus
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Python
jax