3 ms·Flax: A neural network library and ecosystem for Jax designed for flexibility2 points by abracos 10mo ago