Pinned Loading

1. jax-ml/jax jax-ml/jax Public

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

   Python 36.1k 3.7k