Skip to main content
Ctrl+K
torch-harmonics 0.9.2 - Home torch-harmonics 0.9.2 - Home

torch-harmonics 0.9.2

  • GitHub
torch-harmonics 0.9.2 - Home torch-harmonics 0.9.2 - Home

torch-harmonics 0.9.2

  • GitHub

Table of Contents

Getting started

  • Installation
  • Benchmarking
  • Tutorials
    • Getting started with torch-harmonics
    • Visualizing the spherical harmonics
    • Quadrature
    • Polar and azimuthal derivatives
    • Analyzing the gradients of the SHT
    • Helmholtz equation
    • Differentiable shallow water equations
    • Spherical Fourier neural operators
    • Stanford 2D-3D-S dataset
    • Filter basis functions
    • Resampling signals on the 2-sphere
    • Conditioning of the SHT
    • Spherical attention equivariance test

User guide

  • Spherical harmonic transforms
  • Spectral convolutions on the sphere
  • DISCO convolutions on the sphere
  • Attention on the sphere
  • Distributed spherical transforms

API reference

  • Serial layers
    • torch_harmonics.RealSHT
    • torch_harmonics.InverseRealSHT
    • torch_harmonics.RealVectorSHT
    • torch_harmonics.InverseRealVectorSHT
    • torch_harmonics.SpectralConvS2
    • torch_harmonics.DiscreteContinuousConvS2
    • torch_harmonics.DiscreteContinuousConvTransposeS2
    • torch_harmonics.filter_basis.get_filter_basis
    • torch_harmonics.filter_basis.FilterBasis
    • torch_harmonics.filter_basis.PiecewiseLinearFilterBasis
    • torch_harmonics.filter_basis.HarmonicFilterBasis
    • torch_harmonics.filter_basis.ZernikeFilterBasis
    • torch_harmonics.filter_basis.FourierBesselFilterBasis
    • torch_harmonics.AttentionS2
    • torch_harmonics.NeighborhoodAttentionS2
    • torch_harmonics.ResampleS2
    • torch_harmonics.QuadratureS2
    • torch_harmonics.random_fields.GaussianRandomFieldS2
  • Distributed helpers
    • torch_harmonics.distributed.init
    • torch_harmonics.distributed.finalize
    • torch_harmonics.distributed.is_initialized
    • torch_harmonics.distributed.is_distributed_polar
    • torch_harmonics.distributed.is_distributed_azimuth
    • torch_harmonics.distributed.polar_group
    • torch_harmonics.distributed.polar_group_rank
    • torch_harmonics.distributed.polar_group_size
    • torch_harmonics.distributed.azimuth_group
    • torch_harmonics.distributed.azimuth_group_rank
    • torch_harmonics.distributed.azimuth_group_size
    • torch_harmonics.distributed.compute_split_shapes
    • torch_harmonics.distributed.split_tensor_along_dim
  • Distributed layers
    • torch_harmonics.distributed.DistributedRealSHT
    • torch_harmonics.distributed.DistributedInverseRealSHT
    • torch_harmonics.distributed.DistributedRealVectorSHT
    • torch_harmonics.distributed.DistributedInverseRealVectorSHT
    • torch_harmonics.distributed.DistributedSpectralConvS2
    • torch_harmonics.distributed.DistributedDiscreteContinuousConvS2
    • torch_harmonics.distributed.DistributedDiscreteContinuousConvTransposeS2
    • torch_harmonics.distributed.DistributedNeighborhoodAttentionS2
    • torch_harmonics.distributed.DistributedResampleS2
    • torch_harmonics.distributed.DistributedQuadratureS2
  • Distributed primitives
    • torch_harmonics.distributed.distributed_transpose_azimuth
    • torch_harmonics.distributed.distributed_transpose_polar
    • torch_harmonics.distributed.copy_to_polar_region
    • torch_harmonics.distributed.copy_to_azimuth_region
    • torch_harmonics.distributed.reduce_from_polar_region
    • torch_harmonics.distributed.reduce_from_azimuth_region
    • torch_harmonics.distributed.scatter_to_polar_region
    • torch_harmonics.distributed.gather_from_polar_region
    • torch_harmonics.distributed.gather_from_copy_to_polar_region
    • torch_harmonics.distributed.reduce_from_scatter_to_polar_region
    • torch_harmonics.distributed.reduce_from_scatter_to_azimuth_region
    • torch_harmonics.distributed.polar_halo_exchange
    • torch_harmonics.distributed.get_group_neighbors
    • torch_harmonics.distributed.flatten_and_pad_leading_dims
    • torch_harmonics.distributed.unpad_and_unflatten_leading_dims
  • Utilities
    • torch_harmonics.quadrature.precompute_longitudes
    • torch_harmonics.quadrature.precompute_latitudes
    • torch_harmonics.quadrature.legendre_gauss_weights
    • torch_harmonics.quadrature.lobatto_weights
    • torch_harmonics.quadrature.clenshaw_curtiss_weights
    • torch_harmonics.quadrature.trapezoidal_weights
    • torch_harmonics.plotting.plot_sphere
    • torch_harmonics.plotting.imshow_sphere
    • torch_harmonics.truncate_sht

Bibliography

  • References
  • Overview: module code

All modules for which code is available

  • torch_harmonics.attention.attention
  • torch_harmonics.disco.convolution
  • torch_harmonics.distributed.distributed_attention
  • torch_harmonics.distributed.distributed_convolution
  • torch_harmonics.distributed.distributed_quadrature
  • torch_harmonics.distributed.distributed_resample
  • torch_harmonics.distributed.distributed_sht
  • torch_harmonics.distributed.distributed_spectral_convolution
  • torch_harmonics.distributed.primitives
  • torch_harmonics.distributed.utils
  • torch_harmonics.filter_basis
  • torch_harmonics.plotting
  • torch_harmonics.quadrature
  • torch_harmonics.random_fields
  • torch_harmonics.resample
  • torch_harmonics.sht
  • torch_harmonics.spectral_convolution
  • torch_harmonics.truncation
NVIDIA NVIDIA
Privacy Policy | Your Privacy Choices | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2022-2026, The torch-harmonics Authors.