Flax¶
Flax is a high-performance neural network library and ecosystem 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.
homepage: https://flax.readthedocs.io
| version | versionsuffix | toolchain |
|---|---|---|
0.8.4 |
-CUDA-12.1.1 |
gfbf/2023a |
0.8.4 |
gfbf/2023a |
(quick links: (all) - 0 - a - b - c - d - e - f - g - h - i - j - k - l - m - n - o - p - q - r - s - t - u - v - w - x - y - z)