Coax

HuggingFace Transformers

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Trax

Trax — Deep Learning with Clear Code and Speed

Flax

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

Flax NNX

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