Index

A

available_models() (in module neuralop.models.base_model)

B

BasicLoggerCallback (class in neuralop.training.callbacks)

C

Callback (class in neuralop.training.callbacks)

CheckpointCallback (class in neuralop.training.callbacks)

D

DefaultDataProcessor (class in neuralop.datasets.data_transforms)

DomainPadding (class in neuralop.layers.padding)

E

evaluate() (neuralop.training.Trainer method)

F

FNO (class in neuralop.models) FNO1d (class in neuralop.models) FNO2d (class in neuralop.models) FNO3d (class in neuralop.models) forward() (neuralop.datasets.data_transforms.DefaultDataProcessor method) (neuralop.datasets.data_transforms.MGPatchingDataProcessor method) (neuralop.layers.padding.DomainPadding method) (neuralop.layers.skip_connections.SoftGating method) (neuralop.layers.spectral_convolution.SpectralConv method) (neuralop.layers.spectral_convolution.SpectralConv1d method) (neuralop.layers.spectral_convolution.SpectralConv2d method) (neuralop.layers.spectral_convolution.SpectralConv3d method) (neuralop.layers.spherical_convolution.SphericalConv method) (neuralop.models.FNO method) (neuralop.models.SFNO method) (neuralop.models.TFNO method) (neuralop.models.TFNO1d method) (neuralop.models.TFNO2d method) (neuralop.models.TFNO3d method) (neuralop.models.UNO method)

G

get_conv() (neuralop.layers.spectral_convolution.SpectralConv method) (neuralop.layers.spherical_convolution.SphericalConv method)

get_model() (in module neuralop.models.base_model)

L

load_darcy_flow_small() (in module neuralop.datasets)

M

MGPatchingDataProcessor (class in neuralop.datasets.data_transforms) module neuralop.datasets neuralop.datasets.data_transforms neuralop.layers neuralop.layers.padding neuralop.layers.skip_connections neuralop.layers.spectral_convolution neuralop.layers.spherical_convolution neuralop.models neuralop.models.base_model neuralop.training neuralop.training.callbacks

N

neuralop.datasets module neuralop.datasets.data_transforms module neuralop.layers module neuralop.layers.padding module neuralop.layers.skip_connections module neuralop.layers.spectral_convolution module

neuralop.layers.spherical_convolution module neuralop.models module neuralop.models.base_model module neuralop.training module neuralop.training.callbacks module

O

on_epoch_end() (neuralop.training.callbacks.CheckpointCallback method)

on_val_epoch_end() (neuralop.training.callbacks.CheckpointCallback method)

P

pad() (neuralop.layers.padding.DomainPadding method)

postprocess() (neuralop.datasets.data_transforms.MGPatchingDataProcessor method) preprocess() (neuralop.datasets.data_transforms.MGPatchingDataProcessor method)

S

SFNO (class in neuralop.models) skip_connection() (in module neuralop.layers.skip_connections) SoftGating (class in neuralop.layers.skip_connections) SpectralConv (class in neuralop.layers.spectral_convolution)

SpectralConv1d (class in neuralop.layers.spectral_convolution) SpectralConv2d (class in neuralop.layers.spectral_convolution) SpectralConv3d (class in neuralop.layers.spectral_convolution) SphericalConv (class in neuralop.layers.spherical_convolution)

T

TFNO (class in neuralop.models) TFNO1d (class in neuralop.models) TFNO2d (class in neuralop.models) TFNO3d (class in neuralop.models) to() (neuralop.datasets.data_transforms.DefaultDataProcessor method) (neuralop.datasets.data_transforms.MGPatchingDataProcessor method)

train() (neuralop.training.Trainer method) Trainer (class in neuralop.training) transform() (neuralop.layers.spectral_convolution.SpectralConv method) (neuralop.layers.spherical_convolution.SphericalConv method)

U

UNO (class in neuralop.models)

unpad() (neuralop.layers.padding.DomainPadding method)