01
jax-ml/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
- Stars
- 36.2K
- Forks
- 3.7K
- Pushed
- Unknown
Trending repositories
jax-ml’s trending work
Repositories from this owner that have appeared in GitHub Trending.