JAX

JAX: Google's numerical library combining automatic differentiation with just-in-time compilation, favoured for large-scale research on TPUs.

Browse the full glossary →