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