.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/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_sfno.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_resample.py
tests/test_distributed_sht.py
tests/test_sht.py
torch_harmonics/__init__.py
torch_harmonics/_disco_convolution.py
torch_harmonics/_neighborhood_attention.py
torch_harmonics/attention.py
torch_harmonics/cache.py
torch_harmonics/convolution.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.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/csrc/attention/attention.cuh
torch_harmonics/csrc/attention/attention_bwd_cuda.cu
torch_harmonics/csrc/attention/attention_fwd_cuda.cu
torch_harmonics/csrc/attention/attention_interface.cu
torch_harmonics/csrc/attention/attention_utils.cu
torch_harmonics/csrc/attention/attention_utils.cuh
torch_harmonics/csrc/attention/cudamacro.h
torch_harmonics/csrc/disco/disco.h
torch_harmonics/csrc/disco/disco_cuda.cuh
torch_harmonics/csrc/disco/disco_cuda_bwd.cu
torch_harmonics/csrc/disco/disco_cuda_fwd.cu
torch_harmonics/csrc/disco/disco_helpers.cpp
torch_harmonics/csrc/disco/disco_interface.cu
torch_harmonics/distributed/__init__.py
torch_harmonics/distributed/distributed_convolution.py
torch_harmonics/distributed/distributed_resample.py
torch_harmonics/distributed/distributed_sht.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