{
  "$schema": "https://ambiqai.github.io/helia-ui/schema/reference-model-1.json",
  "generatedFrom": {
    "sourceCommit": "f341fb11f7f5d77a4974ba8273c4cd55d67c21f0",
    "tool": "pyref",
    "version": "1.7.3"
  },
  "language": "python",
  "modules": [
    {
      "description": "Typed parameters of the tsmixer family; importable without Keras.",
      "name": "tsmixer_params",
      "path": "helia_edge.models.tsmixer_params",
      "submodules": [],
      "summary": "Typed parameters of the tsmixer family; importable without Keras.",
      "symbols": [
        {
          "description": "TsMixer block parameters",
          "examples": [],
          "id": "helia_edge.models.tsmixer_params.TsMixerBlockParams",
          "kind": "class",
          "language": "python",
          "members": [
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.models.tsmixer_params.TsMixerBlockParams.model_config",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "model_config",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "model_config = ConfigDict(frozen=True, extra='forbid')",
              "source": {
                "line": 18,
                "path": "helia_edge/models/tsmixer_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/models/tsmixer_params.py#L18"
              },
              "summary": ""
            },
            {
              "description": "Normalization type",
              "examples": [],
              "id": "helia_edge.models.tsmixer_params.TsMixerBlockParams.norm",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "norm",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "norm: Literal['batch', 'layer'] = Field(default='layer', description='Normalization type')",
              "source": {
                "line": 20,
                "path": "helia_edge/models/tsmixer_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/models/tsmixer_params.py#L20"
              },
              "summary": ""
            },
            {
              "description": "Activation type",
              "examples": [],
              "id": "helia_edge.models.tsmixer_params.TsMixerBlockParams.activation",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "activation",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "activation: Literal['relu', 'gelu'] = Field(default='relu', description='Activation type')",
              "source": {
                "line": 21,
                "path": "helia_edge/models/tsmixer_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/models/tsmixer_params.py#L21"
              },
              "summary": ""
            },
            {
              "description": "Dropout rate",
              "examples": [],
              "id": "helia_edge.models.tsmixer_params.TsMixerBlockParams.dropout",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "dropout",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "dropout: float = Field(default=None, description='Dropout rate')",
              "source": {
                "line": 22,
                "path": "helia_edge/models/tsmixer_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/models/tsmixer_params.py#L22"
              },
              "summary": ""
            },
            {
              "description": "Feed forward dimension",
              "examples": [],
              "id": "helia_edge.models.tsmixer_params.TsMixerBlockParams.ff_dim",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "ff_dim",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "ff_dim: int = Field(default=None, description='Feed forward dimension')",
              "source": {
                "line": 23,
                "path": "helia_edge/models/tsmixer_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/models/tsmixer_params.py#L23"
              },
              "summary": ""
            }
          ],
          "name": "TsMixerBlockParams",
          "params": [],
          "raises": [],
          "returns": [],
          "signature": "TsMixerBlockParams()",
          "source": {
            "line": 8,
            "path": "helia_edge/models/tsmixer_params.py",
            "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/models/tsmixer_params.py#L8"
          },
          "summary": "TsMixer block parameters"
        },
        {
          "description": "TsMixer parameters",
          "examples": [],
          "id": "helia_edge.models.tsmixer_params.TsMixerParams",
          "kind": "class",
          "language": "python",
          "members": [
            {
              "description": "",
              "examples": [],
              "id": "helia_edge.models.tsmixer_params.TsMixerParams.model_config",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "model_config",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "model_config = ConfigDict(frozen=True, extra='forbid')",
              "source": {
                "line": 36,
                "path": "helia_edge/models/tsmixer_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/models/tsmixer_params.py#L36"
              },
              "summary": ""
            },
            {
              "description": "Model family",
              "examples": [],
              "id": "helia_edge.models.tsmixer_params.TsMixerParams.family",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "family",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "family: Literal['tsmixer'] = 'tsmixer'",
              "source": {
                "line": 38,
                "path": "helia_edge/models/tsmixer_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/models/tsmixer_params.py#L38"
              },
              "summary": ""
            },
            {
              "description": "TsMixer blocks",
              "examples": [],
              "id": "helia_edge.models.tsmixer_params.TsMixerParams.blocks",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "blocks",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "blocks: list[TsBlockParams] = Field(default_factory=list, description='UNext blocks')",
              "source": {
                "line": 39,
                "path": "helia_edge/models/tsmixer_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/models/tsmixer_params.py#L39"
              },
              "summary": ""
            },
            {
              "description": "Output steps; required",
              "examples": [],
              "id": "helia_edge.models.tsmixer_params.TsMixerParams.num_classes",
              "kind": "attribute",
              "language": "python",
              "members": [],
              "name": "num_classes",
              "params": [],
              "raises": [],
              "returns": [],
              "signature": "num_classes: int | None = Field(default=None, gt=0, description='Output steps; required')",
              "source": {
                "line": 40,
                "path": "helia_edge/models/tsmixer_params.py",
                "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/models/tsmixer_params.py#L40"
              },
              "summary": ""
            }
          ],
          "name": "TsMixerParams",
          "params": [],
          "raises": [],
          "returns": [],
          "signature": "TsMixerParams()",
          "source": {
            "line": 26,
            "path": "helia_edge/models/tsmixer_params.py",
            "url": "https://github.com/AmbiqAI/helia-edge/blob/f341fb11f7f5d77a4974ba8273c4cd55d67c21f0/helia_edge/models/tsmixer_params.py#L26"
          },
          "summary": "TsMixer parameters"
        }
      ]
    }
  ],
  "name": "helia_edge"
}