py311-flax
0.6.8_3Neural network library for JAX designed for flexibility
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.
Origin: math/py-flax
Category: math
Size: 2.40MiB
License: APACHE20
Maintainer: sunpoet@FreeBSD.org
Dependencies: 7 packages
Required by: 0 packages
Website: github.com/google/flax
$
pkg install py311-flaxDependencies (7)
More in math
py311-numpy1.26.4_11,1
The New Numeric Extension to PythonR4.5.2
Language for statistical computing and graphicsgmp6.3.0
Free library for arbitrary precision arithmeticopenblas0.3.30,2
Optimized BLAS library based on GotoBLAS2fftw33.3.10_5
Fast C routines to compute the Discrete Fourier Transformpy311-matplotlib3.8.0_2
Plotting library uses a syntax familiar to MATLAB userspy311-pandas2.3.3,1
Flexible, high-performance data analysis in Pythonmpfr4.2.2,1
Library for multiple-precision floating-point computationsoctave10.3.0_2
High-level interactive language for numerical computationsoctave-forge-base1.9_1
Octave-forge baseport for all packages