NVIDIA / torch-harmonicsLinks
Differentiable signal processing on the sphere for PyTorch
☆478Updated 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:
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆253Updated 2 years ago
- A parallel ODE solver for PyTorch☆260Updated 9 months ago
- Newton and Quasi-Newton optimization with PyTorch☆354Updated last month
- ☆264Updated 10 months ago
- Official implementation of NeurIPS 2024 paper "DiffusionPDE: Generative PDE-Solving Under Partial Observation"☆137Updated 2 months ago
- ☆121Updated 7 months ago
- S2FFT: Differentiable and accelerated spherical transforms☆197Updated this week
- Massively parallel training of machine-learning based weather and climate models☆255Updated 2 months ago
- Code for the paper Universal Physics Transformers☆118Updated 2 weeks ago
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆145Updated 3 months ago
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆61Updated last month
- Normalizing flows in PyTorch☆397Updated last month
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆76Updated last month
- code for "Riemannian Flow Matching on General Geometries".☆239Updated last year
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆90Updated 7 months ago
- OSS library that implements deep learning methods for partial differential equations and much more☆445Updated this week
- DeepSphere: a graph-based spherical CNN (PyTorch)☆161Updated 3 years ago
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆164Updated 7 months ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆90Updated this week
- [ICLR2025 Spotlight] Official implementation of Conflict-Free Inverse Gradients Method☆70Updated last month
- Repo to the paper "Message Passing Neural PDE Solvers"☆134Updated 10 months ago
- GPU-Accelerated Radial Basis Function (RBF) Interpolation in PyTorch☆34Updated 2 years ago
- Time- and space-continuous neural PDE forecaster based on INRs and ODEs☆68Updated last year
- ☆158Updated last year
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆375Updated last month
- A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs☆171Updated 10 months ago
- NeuralGCM's differentiable dycore☆282Updated last week
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆72Updated 11 months ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆33Updated last year
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆69Updated 6 months ago