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

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

分支312Tags279

项目介绍

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

定制我的领域
33036.03 K3.68 K访问 GitHub