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

2 points| abracos | 2 months ago |github.com