A pedagogical, functional-oriented deep learning library built on top of jax. - View it on GitHub