LICENSE
README.rst
pyproject.toml
setup.py
neuralop/__init__.py
neuralop/utils.py
neuralop/datasets/__init__.py
neuralop/datasets/burgers.py
neuralop/datasets/darcy.py
neuralop/datasets/data_transforms.py
neuralop/datasets/hdf5_dataset.py
neuralop/datasets/navier_stokes.py
neuralop/datasets/output_encoder.py
neuralop/datasets/pt_dataset.py
neuralop/datasets/spherical_swe.py
neuralop/datasets/tensor_dataset.py
neuralop/datasets/transforms.py
neuralop/datasets/zarr_dataset.py
neuralop/datasets/data/burgers_lowres.pt
neuralop/datasets/data/darcy_test_16.pt
neuralop/datasets/data/darcy_test_32.pt
neuralop/datasets/data/darcy_train_16.pt
neuralop/datasets/tests/__init__.py
neuralop/datasets/tests/test_data_processor.py
neuralop/datasets/tests/test_output_encoder.py
neuralop/layers/__init__.py
neuralop/layers/base_spectral_conv.py
neuralop/layers/einsum_utils.py
neuralop/layers/embeddings.py
neuralop/layers/fno_block.py
neuralop/layers/fourier_continuation.py
neuralop/layers/integral_transform.py
neuralop/layers/legacy_spectral_convolution.py
neuralop/layers/mlp.py
neuralop/layers/neighbor_search.py
neuralop/layers/normalization_layers.py
neuralop/layers/padding.py
neuralop/layers/resample.py
neuralop/layers/segment_csr.py
neuralop/layers/simple_neighbor_search.py
neuralop/layers/skip_connections.py
neuralop/layers/spectral_convolution.py
neuralop/layers/spherical_convolution.py
neuralop/layers/tests/__init__.py
neuralop/layers/tests/test_fno_block.py
neuralop/layers/tests/test_legacy_spectral_convolution.py
neuralop/layers/tests/test_neighbor_search.py
neuralop/layers/tests/test_padding.py
neuralop/layers/tests/test_resample.py
neuralop/layers/tests/test_spectral_convolution.py
neuralop/layers/tests/test_spherical_convolution.py
neuralop/losses/__init__.py
neuralop/losses/data_losses.py
neuralop/losses/equation_losses.py
neuralop/losses/meta_losses.py
neuralop/models/__init__.py
neuralop/models/base_model.py
neuralop/models/fno.py
neuralop/models/fnogno.py
neuralop/models/uno.py
neuralop/models/tests/__init__.py
neuralop/models/tests/test_fno.py
neuralop/models/tests/test_fnogno.py
neuralop/models/tests/test_uno.py
neuralop/mpu/__init__.py
neuralop/mpu/comm.py
neuralop/mpu/helpers.py
neuralop/mpu/mappings.py
neuralop/tests/__init__.py
neuralop/tests/test_model_from_config.py
neuralop/tests/test_utils.py
neuralop/training/__init__.py
neuralop/training/callbacks.py
neuralop/training/load_training_state.py
neuralop/training/patching.py
neuralop/training/torch_setup.py
neuralop/training/trainer.py
neuraloperator.egg-info/PKG-INFO
neuraloperator.egg-info/SOURCES.txt
neuraloperator.egg-info/dependency_links.txt
neuraloperator.egg-info/requires.txt
neuraloperator.egg-info/top_level.txt