Glossary term
Glossary term
Agentic Systems
A programming framework designed for training large-scale neural network models so large that they span multiple TPU accelerator chip slices or pods.
Created for this library
A research lab uses Pax to scale training of its largest language models on TPU pods.
An ML platform team uses Pax as its production training framework for very large transformer models because of its partitioning support.
A research engineer uses Pax to express transformer variants compactly while still scaling efficiently across many TPU chips.
Definition source: Google for Developers Machine Learning Glossary | Creative Commons Attribution 4.0 License