jax-ml/jaxPython
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
(35.767 Stars) (3.610 Forks) (0 indexierte Issues) (0 offene good first issues)
Repositories
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.