Repository info

jax

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