LTSLegendary
jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
- Owner
- jax-ml (organization)
- Stars
- Stars: 36,568
- Forks
- Forks: 4,016
- Language
- Python
- License
- Apache-2.0
- Created
- 2018
- Topics
- jax
- Tier
- LTS (20,000 ★ and up) What this tier means
More repos by jax-ml
All 9 repos by jax-ml- jax-ml/scaling-bookStars: 1,518SCSSBeta
- jax-ml/jax-tritonStars: 473PythonAlpha
- jax-ml/ml_dtypesStars: 366C++Alpha
- jax-ml/oryxStars: 326PythonAlpha
- jax-ml/jax-ai-stackStars: 315PythonAlpha
- jax-ml/jax-llm-examplesStars: 286PythonAlpha
Nearby in Python
All Python repos- satwikkansal/wtfpythonStars: 37,133PythonLTS
- babysor/MockingBirdStars: 36,881PythonLTS
- virgiliojr94/book-to-skillStars: 36,811PythonLTS
- microsoft/graphragStars: 36,536PythonLTS
- hankcs/HanLPStars: 36,532PythonLTS
- XingangPan/DragGANStars: 35,714PythonLTS