BSD Ports

math/py-flax 0.6.8_3 DragonFly

Neural network library for JAX designed for flexibility

System
DragonFly
Repository
DPorts
Maintainer
[email protected] (as in FreeBSD)
License
APACHE20
Categories
math, python
Package size
340.2 KiB, 2454.2 KiB installed

Description

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.

In the other systems

SystemPortVersion
DragonFlymath/py-flax0.6.8_3
FreeBSDmath/py-flax0.6.8_5
HardenedBSDmath/py-flax0.6.8_5
GhostBSDmath/py-flax0.6.8_5

Install

Binary package:

pkg install py311-flax

From DPorts:

cd /usr/dports/math/py-flax && make install clean

Dependencies

Run dependencies 7

No port depends on this one.