Invertible 1D Neural Nets using Haiku and JAX

See this blogpost.