← all repositories

NVIDIA/torch-harmonics

A PyTorch library enabling differentiable signal processing operations on spherical domains, designed for integration into ML training pipelines.

677 stars Jupyter Notebook ML FrameworksDomain Apps
torch-harmonics
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

torch-harmonics implements differentiable spherical harmonic transforms and related signal processing operations within the PyTorch framework. The library provides GPU-accelerated primitives for computing spherical Fourier transforms, convolution operations, and other harmonic analysis tools that maintain gradient flow during backpropagation. These components serve as building blocks for neural networks operating on spherical data such as global climate data, molecular structures, or 3D point clouds.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.