Flax: A neural network library and ecosystem for Jax designed for flexibility

No comments yet.