JAX

High-performance machine learning research library with composable transformations of Python functions.

Explore on Agidapp Global