.clang-format
.clang-format-ignore
.coveragerc
.gitattributes
.gitignore
AUTHORS
Changelog.md
Dockerfile
Dockerfile.examples
LICENSE
README.md
pyproject.toml
setup.py
.github/dependabot.yml
.github/workflows/deploy_pypi.yml
.github/workflows/tests.yml
examples/model_registry.py
examples/baseline_models/__init__.py
examples/baseline_models/segformer.py
examples/baseline_models/transformer.py
examples/baseline_models/unet.py
examples/depth/train.py
examples/segmentation/train.py
examples/shallow_water_equations/train.py
images/allen-cahn.gif
images/ginzburg-landau.gif
images/sfno.gif
images/spherical_harmonics.gif
images/zonal_jet.gif
images/logo/logo.png
notebooks/conditioning_sht.ipynb
notebooks/equivariance_test.ipynb
notebooks/getting_started.ipynb
notebooks/gradient_analysis.ipynb
notebooks/helmholtz.ipynb
notebooks/partial_derivatives.ipynb
notebooks/plot_spherical_harmonics.ipynb
notebooks/quadrature.ipynb
notebooks/resample_sphere.ipynb
notebooks/shallow_water_equations.ipynb
notebooks/stanford_2d3ds_dataset.ipynb
notebooks/train_spherical_neural_operator.ipynb
tests/run_tests.sh
tests/test_attention.py
tests/test_cache.py
tests/test_convolution.py
tests/test_distributed_convolution.py
tests/test_distributed_primitives.py
tests/test_distributed_quadrature.py
tests/test_distributed_resample.py
tests/test_distributed_sht.py
tests/test_distributed_spectral_convolution.py
tests/test_quadrature.py
tests/test_random_fields.py
tests/test_resample.py
tests/test_sht.py
tests/test_spectral_convolution.py
tests/testutils.py
torch_harmonics/__init__.py
torch_harmonics/cache.py
torch_harmonics/fft.py
torch_harmonics/filter_basis.py
torch_harmonics/legendre.py
torch_harmonics/plotting.py
torch_harmonics/quadrature.py
torch_harmonics/random_fields.py
torch_harmonics/resample.py
torch_harmonics/sht.py
torch_harmonics/spectral_convolution.py
torch_harmonics/truncation.py
torch_harmonics.egg-info/PKG-INFO
torch_harmonics.egg-info/SOURCES.txt
torch_harmonics.egg-info/dependency_links.txt
torch_harmonics.egg-info/requires.txt
torch_harmonics.egg-info/top_level.txt
torch_harmonics/attention/__init__.py
torch_harmonics/attention/_attention_utils.py
torch_harmonics/attention/attention.py
torch_harmonics/attention/csrc/attention.h
torch_harmonics/attention/csrc/attention_cpu.h
torch_harmonics/attention/csrc/attention_cpu_bwd.cpp
torch_harmonics/attention/csrc/attention_cpu_fwd.cpp
torch_harmonics/attention/csrc/attention_cuda.cuh
torch_harmonics/attention/csrc/attention_cuda_bwd.cu
torch_harmonics/attention/csrc/attention_cuda_fwd.cu
torch_harmonics/attention/csrc/attention_cuda_utils.cu
torch_harmonics/attention/csrc/attention_cuda_utils.cuh
torch_harmonics/attention/csrc/attention_helpers.cpp
torch_harmonics/attention/csrc/attention_interface.cpp
torch_harmonics/attention/csrc/attention_interface.cu
torch_harmonics/attention/csrc/cudamacro.h
torch_harmonics/disco/__init__.py
torch_harmonics/disco/_disco_utils.py
torch_harmonics/disco/convolution.py
torch_harmonics/disco/csrc/disco.h
torch_harmonics/disco/csrc/disco_cpu.cpp
torch_harmonics/disco/csrc/disco_cpu.h
torch_harmonics/disco/csrc/disco_cuda.cuh
torch_harmonics/disco/csrc/disco_cuda_bwd.cu
torch_harmonics/disco/csrc/disco_cuda_fwd.cu
torch_harmonics/disco/csrc/disco_helpers.cpp
torch_harmonics/disco/csrc/disco_helpers.h
torch_harmonics/disco/csrc/disco_interface.cpp
torch_harmonics/distributed/__init__.py
torch_harmonics/distributed/distributed_convolution.py
torch_harmonics/distributed/distributed_quadrature.py
torch_harmonics/distributed/distributed_resample.py
torch_harmonics/distributed/distributed_sht.py
torch_harmonics/distributed/distributed_spectral_convolution.py
torch_harmonics/distributed/primitives.py
torch_harmonics/distributed/utils.py
torch_harmonics/examples/__init__.py
torch_harmonics/examples/losses.py
torch_harmonics/examples/metrics.py
torch_harmonics/examples/pde_dataset.py
torch_harmonics/examples/pde_sphere.py
torch_harmonics/examples/shallow_water_equations.py
torch_harmonics/examples/stanford_2d3ds_dataset.py
torch_harmonics/examples/models/__init__.py
torch_harmonics/examples/models/_layers.py
torch_harmonics/examples/models/lsno.py
torch_harmonics/examples/models/s2segformer.py
torch_harmonics/examples/models/s2transformer.py
torch_harmonics/examples/models/s2unet.py
torch_harmonics/examples/models/sfno.py