GitHub repository: mahdi-shafiei/awesome-jax
Description: JAX - A curated list of resources https://github.com/google/jax