📈 GH Trending Explorer 🎲 Surprise me

Search: vectorize · Language: Python · clear

jax-ml / jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more