Index

A

abs() (neuralop.losses.H1Loss method) (neuralop.losses.LpLoss method)

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

C

CarCFDDataset (class in neuralop.data.datasets.car_cfd_dataset) checkpoint() (neuralop.training.Trainer method)

CODABlocks (class in neuralop.layers.coda_blocks) compute_attention() (neuralop.layers.coda_blocks.CODABlocks method) compute_terms() (neuralop.losses.H1Loss method)

D

DarcyDataset (class in neuralop.data.datasets.darcy) DefaultDataProcessor (class in neuralop.data.transforms.data_processors)

DiscreteContinuousConv2d (class in neuralop.layers.discrete_continuous_convolution) DiscreteContinuousConvTranspose2d (class in neuralop.layers.discrete_continuous_convolution) DomainPadding (class in neuralop.layers.padding)

E

EquidistantDiscreteContinuousConv2d (class in neuralop.layers.discrete_continuous_convolution) EquidistantDiscreteContinuousConvTranspose2d (class in neuralop.layers.discrete_continuous_convolution)

eval_one_batch() (neuralop.training.Trainer method) evaluate() (neuralop.training.Trainer method)

F

FiniteDifferenceConvolution (class in neuralop.layers.differential_conv) FNO (class in neuralop.models) FNO1d (class in neuralop.models) FNO2d (class in neuralop.models) FNO3d (class in neuralop.models) forward() (neuralop.data.transforms.data_processors.DefaultDataProcessor method) (neuralop.data.transforms.data_processors.MGPatchingDataProcessor method) (neuralop.layers.coda_blocks.CODABlocks method) (neuralop.layers.differential_conv.FiniteDifferenceConvolution method) (neuralop.layers.discrete_continuous_convolution.DiscreteContinuousConv2d method) (neuralop.layers.discrete_continuous_convolution.DiscreteContinuousConvTranspose2d method) (neuralop.layers.discrete_continuous_convolution.EquidistantDiscreteContinuousConv2d method) (neuralop.layers.discrete_continuous_convolution.EquidistantDiscreteContinuousConvTranspose2d method) (neuralop.layers.embeddings.GridEmbedding2D method) (neuralop.layers.embeddings.GridEmbeddingND method) (neuralop.layers.embeddings.SinusoidalEmbedding method) (neuralop.layers.gno_block.GNOBlock method) (neuralop.layers.integral_transform.IntegralTransform method) (neuralop.layers.local_fno_block.LocalFNOBlocks method) (neuralop.layers.neighbor_search.NeighborSearch 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.GINO 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_block() (neuralop.layers.local_fno_block.LocalFNOBlocks method) get_local_filter_matrix() (neuralop.layers.discrete_continuous_convolution.DiscreteContinuousConv2d method) (neuralop.layers.discrete_continuous_convolution.DiscreteContinuousConvTranspose2d method) (neuralop.layers.discrete_continuous_convolution.EquidistantDiscreteContinuousConv2d method) (neuralop.layers.discrete_continuous_convolution.EquidistantDiscreteContinuousConvTranspose2d method) get_model() (in module neuralop.models.base_model)

GINO (class in neuralop.models) GNOBlock (class in neuralop.layers.gno_block) grid() (neuralop.layers.embeddings.GridEmbedding2D method) (neuralop.layers.embeddings.GridEmbeddingND method) GridEmbedding2D (class in neuralop.layers.embeddings) GridEmbeddingND (class in neuralop.layers.embeddings)

H

H1Loss (class in neuralop.losses)

I

IncrementalFNOTrainer (class in neuralop.training)

IntegralTransform (class in neuralop.layers.integral_transform)

L

load_darcy_flow_small() (in module neuralop.data.datasets) LocalFNOBlocks (class in neuralop.layers.local_fno_block) log_eval() (neuralop.training.Trainer method)

log_training() (neuralop.training.Trainer method) loss_gap() (neuralop.training.IncrementalFNOTrainer method) LpLoss (class in neuralop.losses)

M

MGPatchingDataProcessor (class in neuralop.data.transforms.data_processors) module neuralop.data.datasets neuralop.data.datasets.car_cfd_dataset neuralop.data.datasets.darcy neuralop.data.datasets.navier_stokes neuralop.data.transforms.data_processors neuralop.layers neuralop.layers.coda_blocks neuralop.layers.differential_conv neuralop.layers.discrete_continuous_convolution neuralop.layers.embeddings, [1] neuralop.layers.gno_block neuralop.layers.integral_transform neuralop.layers.local_fno_block neuralop.layers.neighbor_search neuralop.layers.padding neuralop.layers.skip_connections neuralop.layers.spectral_convolution neuralop.layers.spherical_convolution neuralop.losses neuralop.models neuralop.models.base_model neuralop.training

MSELoss (class in neuralop.losses)

N

native_neighbor_search() (in module neuralop.layers.neighbor_search) NavierStokesDataset (class in neuralop.data.datasets.navier_stokes) NeighborSearch (class in neuralop.layers.neighbor_search) neuralop.data.datasets module neuralop.data.datasets.car_cfd_dataset module neuralop.data.datasets.darcy module neuralop.data.datasets.navier_stokes module neuralop.data.transforms.data_processors module neuralop.layers module neuralop.layers.coda_blocks module neuralop.layers.differential_conv module neuralop.layers.discrete_continuous_convolution module neuralop.layers.embeddings module, [1]

neuralop.layers.gno_block module neuralop.layers.integral_transform module neuralop.layers.local_fno_block module neuralop.layers.neighbor_search module neuralop.layers.padding module neuralop.layers.skip_connections module neuralop.layers.spectral_convolution module neuralop.layers.spherical_convolution module neuralop.losses module neuralop.models module neuralop.models.base_model module neuralop.training module

O

on_epoch_start() (neuralop.training.Trainer method)

out_channels (neuralop.layers.embeddings.SinusoidalEmbedding property)

P

pad() (neuralop.layers.padding.DomainPadding method) postprocess() (neuralop.data.transforms.data_processors.DefaultDataProcessor method) (neuralop.data.transforms.data_processors.MGPatchingDataProcessor method)

preprocess() (neuralop.data.transforms.data_processors.DefaultDataProcessor method) (neuralop.data.transforms.data_processors.MGPatchingDataProcessor method)

R

reduce_all() (neuralop.losses.H1Loss method) (neuralop.losses.LpLoss method)

rel() (neuralop.losses.H1Loss method) (neuralop.losses.LpLoss method) resume_state_from_dir() (neuralop.training.Trainer method)

S

set_ada_in_embeddings() (neuralop.layers.local_fno_block.LocalFNOBlocks method) SFNO (class in neuralop.models) SinusoidalEmbedding (class in neuralop.layers.embeddings) 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.data.transforms.data_processors.DefaultDataProcessor method) (neuralop.data.transforms.data_processors.MGPatchingDataProcessor method)

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

U

uniform_h() (neuralop.losses.H1Loss method) (neuralop.losses.LpLoss method)

UNO (class in neuralop.models) unpad() (neuralop.layers.padding.DomainPadding method)