airfRANS-model-exploration/src/airfrans_frontier/models/__init__.py

"""Baseline model definitions."""

from airfrans_frontier.models.coordinate_encoding import CoordinateEncoder, CoordinateEncodingSpec
from airfrans_frontier.models.film import FourierFiLMMLP
from airfrans_frontier.models.frontier import (
    DeepONetBranchTrunk,
    LocalPointTransformer,
    NeRFCFDMultiRes,
    PointContextPerceiver,
    RasterFNOUNet,
    SirenConditionedINR,
)
from airfrans_frontier.models.mlp import EncodedPointwiseMLP, PointwiseMLP

__all__ = [
    "CoordinateEncoder",
    "CoordinateEncodingSpec",
    "DeepONetBranchTrunk",
    "EncodedPointwiseMLP",
    "FourierFiLMMLP",
    "LocalPointTransformer",
    "NeRFCFDMultiRes",
    "PointContextPerceiver",
    "PointwiseMLP",
    "RasterFNOUNet",
    "SirenConditionedINR",
]