jax-ml/jax

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

语言：Python · GitHub 星标：36,582 · Fork：3,989

话题：jax

在 GitHub 上查看原仓库