Flax: A neural network library and ecosystem for Jax designed for flexibility github.com 1 points by abracos 12 hours ago