Repository info

jax

• Summary: a summary of jax
• Description: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
• Stars: 33314
• Number of forks: 3146