Index

A

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

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

C

checkpoint() (neuralop.training.Trainer method)

compute_terms() (neuralop.losses.H1Loss method)

D

DefaultDataProcessor (class in neuralop.data.transforms.data_processors)

DomainPadding (class in neuralop.layers.padding)

E

eval_one_batch() (neuralop.training.Trainer method)

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.data.transforms.data_processors.DefaultDataProcessor method) (neuralop.data.transforms.data_processors.MGPatchingDataProcessor 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.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_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) 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.transforms.data_processors neuralop.layers neuralop.layers.embeddings, [1] neuralop.layers.gno_block neuralop.layers.integral_transform 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) NeighborSearch (class in neuralop.layers.neighbor_search) neuralop.data.datasets module neuralop.data.transforms.data_processors module neuralop.layers module neuralop.layers.embeddings module, [1] neuralop.layers.gno_block module neuralop.layers.integral_transform 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

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)