NVIDIA / torch-harmonicsLinks
Differentiable signal processing on the sphere for PyTorch
☆518Updated this week
Alternatives and similar repositories for torch-harmonics
Users that are interested in torch-harmonics are comparing it to the libraries listed below
Sorting:
- A parallel ODE solver for PyTorch☆264Updated 10 months ago
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆261Updated 2 years ago
- Newton and Quasi-Newton optimization with PyTorch☆363Updated 3 months ago
- ☆269Updated 11 months ago
- ☆125Updated 8 months ago
- Massively parallel training of machine-learning based weather and climate models☆310Updated last week
- Code for the paper Universal Physics Transformers☆127Updated 2 months ago
- S2FFT: Differentiable and accelerated spherical transforms☆202Updated this week
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆172Updated 8 months ago
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆151Updated 4 months ago
- OSS library that implements deep learning methods for partial differential equations and much more☆450Updated last month
- Normalizing flows in PyTorch☆402Updated 2 months ago
- PyTorch implementation of Levenberg-Marquardt training algorithm☆75Updated 5 months ago
- ☆163Updated last year
- Official implementation of NeurIPS 2024 paper "DiffusionPDE: Generative PDE-Solving Under Partial Observation"☆144Updated 4 months ago
- GPU-Accelerated Radial Basis Function (RBF) Interpolation in PyTorch☆37Updated last month
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆90Updated 3 weeks ago
- A Python library for end-to-end learning on surfaces. It implements pre-processing functions that include geodesic algorithms, neural net…☆34Updated 2 weeks ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆207Updated 3 weeks ago
- code for "Riemannian Flow Matching on General Geometries".☆255Updated last year
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆81Updated 3 months ago
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆386Updated 3 weeks ago
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆65Updated 2 months ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆48Updated 3 months ago
- [ICLR2025 Spotlight] Official implementation of Conflict-Free Inverse Gradients Method☆78Updated 3 months ago
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆93Updated 8 months ago
- TorchFSM: Fourier Spectral Method with PyTorch☆48Updated 2 weeks ago
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆463Updated 10 months ago
- Repo to the paper "Message Passing Neural PDE Solvers"☆135Updated 11 months ago
- Time- and space-continuous neural PDE forecaster based on INRs and ODEs☆71Updated last year