ports/math/py-flax/pkg-descr
Sunpoet Po-Chuan Hsieh 73be298892 Add py-flax 0.1.0
Flax is a high-performance neural network library for JAX that is designed for
flexibility: Try new forms of training by forking an example and by modifying
the training loop, not by adding features to a framework.

WWW: https://github.com/google/flax
2020-07-09 18:08:11 +00:00

5 lines
252 B
Text

Flax is a high-performance neural network library for JAX that is designed for
flexibility: Try new forms of training by forking an example and by modifying
the training loop, not by adding features to a framework.
WWW: https://github.com/google/flax