API Reference

gaia package
gaia
- gaia package

Core Components

gaia.core package
gaia.models package
- create_gaia_llm()
- Submodules
  - gaia.models.categorical_mlp module
  - gaia.models.gaia_transformer module
    - create_gaia_llm()
gaia.nn module
gaia.data package
- Submodules
gaia.integration package

Training & Optimization

gaia.training package
- Subpackages
- Submodules
gaia.callbacks package
- CallbackManager
- EarlyStopping
  - EarlyStopping.__init__()
  - EarlyStopping.on_epoch_end()

Utilities

gaia.utils package
- set_seed()
- Submodules
  - gaia.utils.device module
  - gaia.utils.reproducibility module
    - set_seed()
gaia.pytorch_api module

GAIA Framework

gaia package
gaia.training package
gaia.training.engine package
gaia.training.engine.profiler module

gaia.training.engine.profiler module

Previous Next

© Copyright 2025, Juan Zambrano, Sridhar Mahadevan, Enrique ter Horst. Last updated on Sep 05, 2025.