Skip to content

framework

Flax

Neural network library built on top of JAX, providing modules and training utilities for machine learning models.

Current clusters