Titan AI LogoTitan AI

flax

6,807
736
Jupyter Notebook

Project Description

Flax is a neural network library for JAX that is designed for flexibility.

Project Information

Created on 1/10/2020
Updated on 9/18/2025