Praxis

A core, high-performance ML library of Pax. Praxis is often called the "Layer library".

Praxis contains not just the definitions for the Layer class, but most of its supporting components as well, including:

data inputs

configuration libraries (HParam and Fiddle)

Praxis provides the definitions for the Model class.

Created for this library

1.
A research lab uses Praxis as the layer library underlying Pax for its transformer experiments.
2.
An ML platform team uses Praxis components to assemble transformer architectures consistently across teams.
3.
A research team uses Praxis to build a custom transformer variant while reusing Pax training infrastructure.

Loading…