gaia.models package

GAIA Model Architectures

gaia.models.create_gaia_llm(vocab_size, model_size='base', use_all_gaia_features=True)

Create a GAIA LLM with predefined configurations

Parameters:

• vocab_size (int) – Vocabulary size

• model_size (str) – ‘small’, ‘base’, ‘large’, or ‘xl’

• use_all_gaia_features (bool) – Whether to use all GAIA enhancements

Returns:

Configured GAIATransformer

Return type:

<MagicMock name=’mock.__getitem__()’ id=’4541362384’>

Submodules

• gaia.models.categorical_mlp module
• gaia.models.gaia_transformer module
  • create_gaia_llm()