Titan AI LogoTitan AI

flax

6,897
753
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 11/6/2025