An Open Source Machine Learning Framework for Everyone

Democratizing Reinforcement Learning for LLMs

Lingvo

Reference models and tools for Cloud TPUs.

Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry lead…