jax-ml/jaxPython
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
(35,767 stars) (3,610 forks) (0 个已索引 issue) (0 个开放 good first issue)
仓库
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
A stand-alone implementation of several NumPy dtype extensions used in machine learning.