Index

A

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

AdaIN (class in neuralop.layers.normalization_layers) autograd() (neuralop.losses.equation_losses.PoissonInteriorLoss method) available_models() (in module neuralop.models.base_model)

B

BatchNorm (class in neuralop.layers.normalization_layers)

Burgers1dTimeDataset (class in neuralop.data.datasets.burgers) BurgersEqnLoss (class in neuralop.losses.equation_losses)

C

CarCFDDataset (class in neuralop.data.datasets.car_cfd_dataset) ChannelMLP (class in neuralop.layers.channel_mlp) checkpoint() (neuralop.training.Trainer method) CODALayer (class in neuralop.layers.coda_layer) CODANO (class in neuralop.models) ComplexValued (class in neuralop.layers.complex) compute_attention() (neuralop.layers.coda_layer.CODALayer method) compute_explained_variance() (in module neuralop.utils) compute_multiple_derivatives() (neuralop.losses.differentiation.FourierDiff method)

compute_rank() (in module neuralop.utils) compute_stable_rank() (in module neuralop.utils) compute_terms() (neuralop.losses.data_losses.H1Loss method) (neuralop.losses.data_losses.HdivLoss method) count_model_params() (in module neuralop.utils) count_tensor_params() (in module neuralop.utils) cpu() (neuralop.data.transforms.normalizers.UnitGaussianNormalizer method) cuda() (neuralop.data.transforms.normalizers.UnitGaussianNormalizer method) curl() (neuralop.losses.differentiation.FiniteDiff method) (neuralop.losses.differentiation.FourierDiff method)

D

DarcyDataset (class in neuralop.data.datasets.darcy) DefaultDataProcessor (class in neuralop.data.transforms.data_processors) derivative() (neuralop.losses.differentiation.FourierDiff method) DictUnitGaussianNormalizer (class in neuralop.data.transforms.normalizers) DiscreteContinuousConv2d (class in neuralop.layers.discrete_continuous_convolution) DiscreteContinuousConvTranspose2d (class in neuralop.layers.discrete_continuous_convolution) divergence() (neuralop.losses.differentiation.FiniteDiff method) (neuralop.losses.differentiation.FourierDiff method)

DomainPadding (class in neuralop.layers.padding) dx() (neuralop.losses.differentiation.FiniteDiff method) (neuralop.losses.differentiation.FourierDiff method) dy() (neuralop.losses.differentiation.FiniteDiff method) (neuralop.losses.differentiation.FourierDiff method) dz() (neuralop.losses.differentiation.FiniteDiff method) (neuralop.losses.differentiation.FourierDiff method)

E

EquidistantDiscreteContinuousConv2d (class in neuralop.layers.discrete_continuous_convolution) EquidistantDiscreteContinuousConvTranspose2d (class in neuralop.layers.discrete_continuous_convolution) eval_one_batch() (neuralop.training.Trainer method)

eval_one_batch_autoreg() (neuralop.training.Trainer method) evaluate() (neuralop.training.Trainer method) evaluate_all() (neuralop.training.Trainer method)

F

FiniteDiff (class in neuralop.losses.differentiation) FiniteDifferenceConvolution (class in neuralop.layers.differential_conv) FNO (class in neuralop.models) FNOBlocks (class in neuralop.layers.fno_block) FNOGNO (class in neuralop.models) forward() (neuralop.data.transforms.data_processors.DefaultDataProcessor method) (neuralop.data.transforms.data_processors.IncrementalDataProcessor method) (neuralop.data.transforms.data_processors.MGPatchingDataProcessor method) (neuralop.data.transforms.normalizers.UnitGaussianNormalizer method) (neuralop.layers.channel_mlp.ChannelMLP method) (neuralop.layers.coda_layer.CODALayer method) (neuralop.layers.complex.ComplexValued 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.fno_block.FNOBlocks method) (neuralop.layers.gno_block.GNOBlock method) (neuralop.layers.integral_transform.IntegralTransform method) (neuralop.layers.local_no_block.LocalNOBlocks method) (neuralop.layers.neighbor_search.NeighborSearch method) (neuralop.layers.normalization_layers.AdaIN method) (neuralop.layers.normalization_layers.BatchNorm method) (neuralop.layers.normalization_layers.InstanceNorm method) (neuralop.layers.padding.DomainPadding method) (neuralop.layers.spectral_convolution.SpectralConv method) (neuralop.layers.spherical_convolution.SphericalConv method) (neuralop.losses.meta_losses.Relobralo method) (neuralop.losses.meta_losses.SoftAdapt method) (neuralop.models.CODANO method) (neuralop.models.FNO method) (neuralop.models.FNOGNO method) (neuralop.models.GINO method) (neuralop.models.LocalNO method) (neuralop.models.SFNO method) (neuralop.models.UNO method) (neuralop.models.UQNO method)

FourierDiff (class in neuralop.losses.differentiation) from_dataset() (neuralop.data.transforms.normalizers.DictUnitGaussianNormalizer class method) (neuralop.data.transforms.normalizers.UnitGaussianNormalizer class method)

G

get_block() (neuralop.layers.fno_block.FNOBlocks method) (neuralop.layers.local_no_block.LocalNOBlocks 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) grad_explained() (neuralop.training.IncrementalFNOTrainer method) gradient() (neuralop.losses.differentiation.FiniteDiff method) (neuralop.losses.differentiation.FourierDiff method) 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.data_losses)

HdivLoss (class in neuralop.losses.data_losses)

I

ICLoss (class in neuralop.losses.equation_losses) incremental_update() (neuralop.training.IncrementalFNOTrainer method) IncrementalDataProcessor (class in neuralop.data.transforms.data_processors)

IncrementalFNOTrainer (class in neuralop.training) InstanceNorm (class in neuralop.layers.normalization_layers) IntegralTransform (class in neuralop.layers.integral_transform) integrate_latent() (neuralop.models.FNOGNO method)

L

laplacian() (neuralop.losses.differentiation.FiniteDiff method) (neuralop.losses.differentiation.FourierDiff method) load_darcy_flow_small() (in module neuralop.data.datasets) load_mini_burgers_1dtime() (in module neuralop.data.datasets.burgers) LocalNO (class in neuralop.models)

LocalNOBlocks (class in neuralop.layers.local_no_block) log_eval() (neuralop.training.Trainer method) log_training() (neuralop.training.Trainer method) loss_gap() (neuralop.training.IncrementalFNOTrainer method) LpLoss (class in neuralop.losses.data_losses)

M

MGPatchingDataProcessor (class in neuralop.data.transforms.data_processors) module neuralop.data.datasets neuralop.data.datasets.burgers neuralop.data.datasets.car_cfd_dataset neuralop.data.datasets.darcy neuralop.data.datasets.navier_stokes neuralop.data.transforms.data_processors neuralop.data.transforms.normalizers neuralop.layers neuralop.layers.channel_mlp neuralop.layers.coda_layer neuralop.layers.complex neuralop.layers.differential_conv neuralop.layers.discrete_continuous_convolution neuralop.layers.embeddings neuralop.layers.fno_block neuralop.layers.gno_block neuralop.layers.integral_transform neuralop.layers.local_no_block neuralop.layers.neighbor_search neuralop.layers.normalization_layers neuralop.layers.padding neuralop.layers.skip_connections neuralop.layers.spectral_convolution neuralop.layers.spectral_projection neuralop.layers.spherical_convolution neuralop.losses.data_losses neuralop.losses.differentiation neuralop.losses.equation_losses neuralop.losses.meta_losses neuralop.models neuralop.models.base_model neuralop.training neuralop.training.patching neuralop.training.torch_setup neuralop.utils

MSELoss (class in neuralop.losses.data_losses) MultigridPatching2D (class in neuralop.training.patching)

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.burgers 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.data.transforms.normalizers module neuralop.layers module neuralop.layers.channel_mlp module neuralop.layers.coda_layer module neuralop.layers.complex module neuralop.layers.differential_conv module neuralop.layers.discrete_continuous_convolution module neuralop.layers.embeddings module neuralop.layers.fno_block module neuralop.layers.gno_block module neuralop.layers.integral_transform module

neuralop.layers.local_no_block module neuralop.layers.neighbor_search module neuralop.layers.normalization_layers module neuralop.layers.padding module neuralop.layers.skip_connections module neuralop.layers.spectral_convolution module neuralop.layers.spectral_projection module neuralop.layers.spherical_convolution module neuralop.losses.data_losses module neuralop.losses.differentiation module neuralop.losses.equation_losses module neuralop.losses.meta_losses module neuralop.models module neuralop.models.base_model module neuralop.training module neuralop.training.patching module neuralop.training.torch_setup module neuralop.utils module non_uniform_fd() (in module neuralop.losses.differentiation)

O

on_epoch_start() (neuralop.training.Trainer method)

out_channels (neuralop.layers.embeddings.SinusoidalEmbedding property)

P

pad() (neuralop.layers.padding.DomainPadding method) partial() (neuralop.losses.differentiation.FourierDiff method) patch() (neuralop.training.patching.MultigridPatching2D method) PointwiseQuantileLoss (class in neuralop.losses.data_losses) PoissonBoundaryLoss (class in neuralop.losses.equation_losses)

PoissonEqnLoss (class in neuralop.losses.equation_losses) PoissonInteriorLoss (class in neuralop.losses.equation_losses) 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.data_losses.H1Loss method) (neuralop.losses.data_losses.HdivLoss method) (neuralop.losses.data_losses.LpLoss method) (neuralop.losses.data_losses.PointwiseQuantileLoss method)

rel() (neuralop.losses.data_losses.H1Loss method) (neuralop.losses.data_losses.HdivLoss method) (neuralop.losses.data_losses.LpLoss method) Relobralo (class in neuralop.losses.meta_losses) resume_state_from_dir() (neuralop.training.Trainer method)

S

set_ada_in_embeddings() (neuralop.layers.fno_block.FNOBlocks method) (neuralop.layers.local_no_block.LocalNOBlocks method) set_embedding() (neuralop.layers.normalization_layers.AdaIN method) setup() (in module neuralop.training.torch_setup) SFNO (class in neuralop.models) SinusoidalEmbedding (class in neuralop.layers.embeddings)

skip_connection() (in module neuralop.layers.skip_connections) SoftAdapt (class in neuralop.losses.meta_losses) spectral_projection_divergence_free() (in module neuralop.layers.spectral_projection) SpectralConv (class in neuralop.layers.spectral_convolution) spectrum_2d() (in module neuralop.utils) SphericalConv (class in neuralop.layers.spherical_convolution)

T

TFNO (class in neuralop.models) to() (neuralop.data.transforms.data_processors.DefaultDataProcessor method) (neuralop.data.transforms.data_processors.IncrementalDataProcessor method) (neuralop.data.transforms.data_processors.MGPatchingDataProcessor method) (neuralop.data.transforms.normalizers.UnitGaussianNormalizer 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_quadrature() (neuralop.losses.data_losses.H1Loss method) (neuralop.losses.data_losses.HdivLoss method) (neuralop.losses.data_losses.LpLoss method) UnitGaussianNormalizer (class in neuralop.data.transforms.normalizers)

UNO (class in neuralop.models) unpad() (neuralop.layers.padding.DomainPadding method) unpatch() (neuralop.training.patching.MultigridPatching2D method) UQNO (class in neuralop.models)

W

WeightedSumLoss (class in neuralop.losses.meta_losses)