JAX - A curated list of resources https://github.com/google/jax - View it on GitHub