JAX - A curated list of resources https://github.com/google/jax
Awesome resources on normalizing flows.