.coveragerc
.gitattributes
.gitignore
AUTHORS
Changelog.md
Dockerfile
LICENSE
README.md
pyproject.toml
setup.py
.github/dependabot.yml
.github/workflows/deploy_pypi.yml
.github/workflows/tests.yml
examples/minimal_example.py
examples/train_sfno.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/plot_spherical_harmonics.ipynb
notebooks/plotting.py
notebooks/quadrature.ipynb
notebooks/shallow_water_equations.ipynb
notebooks/train_sfno.ipynb
tests/test_convolution.py
tests/test_distributed_convolution.py
tests/test_distributed_sht.py
tests/test_sht.py
torch_harmonics/__init__.py
torch_harmonics/_disco_convolution.py
torch_harmonics/convolution.py
torch_harmonics/legendre.py
torch_harmonics/quadrature.py
torch_harmonics/random_fields.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/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_sht.py
torch_harmonics/distributed/primitives.py
torch_harmonics/distributed/utils.py
torch_harmonics/examples/__init__.py
torch_harmonics/examples/pde_sphere.py
torch_harmonics/examples/shallow_water_equations.py
torch_harmonics/examples/sfno/__init__.py
torch_harmonics/examples/sfno/models/__init__.py
torch_harmonics/examples/sfno/models/activations.py
torch_harmonics/examples/sfno/models/contractions.py
torch_harmonics/examples/sfno/models/factorizations.py
torch_harmonics/examples/sfno/models/layers.py
torch_harmonics/examples/sfno/models/sfno.py
torch_harmonics/examples/sfno/utils/pde_dataset.py