Stars
6
results
for source starred repositories
written in Python
Clear filter
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Interactive deep learning book with multi-framework code, math, and discussions. Adopted at 500 universities from 70 countries including Stanford, MIT, Harvard, and Cambridge.
Scenic: A Jax Library for Computer Vision Research and Beyond
Awesome Reasoning LLM Tutorial/Survey/Guide
Optax is a gradient processing and optimization library for JAX.
Library for reading and processing ML training data.