jax-ml/jaxPython
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
(36.162 star) (3735 fork) (4 issue indicizzate) (4 good first issue aperte)
Repository
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.
Oryx is a library for probabilistic programming and deep learning built on top of Jax.