NVIDIA / torch-harmonicsLinks
Differentiable signal processing on the sphere for PyTorch
☆627Updated last 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☆277Updated last year
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆275Updated 3 years ago
- ☆287Updated last year
- Massively parallel training of machine-learning based weather and climate models☆351Updated last week
- Newton and Quasi-Newton optimization with PyTorch☆372Updated last week
- Code for the paper Universal Physics Transformers☆149Updated 6 months ago
- ☆130Updated last week
- S2FFT: Differentiable and accelerated spherical transforms☆215Updated this week
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆169Updated 9 months ago
- Official implementation of NeurIPS 2024 paper "DiffusionPDE: Generative PDE-Solving Under Partial Observation"☆160Updated 9 months ago
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆187Updated last year
- Normalizing flows in PyTorch☆441Updated last month
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆96Updated 2 months ago
- Official implementation of "Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulation"☆51Updated 3 months ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆52Updated 8 months ago
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆75Updated 7 months ago
- ☆173Updated 2 years ago
- OSS library that implements deep learning methods for partial differential equations and much more☆458Updated 3 months ago
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆102Updated last year
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆434Updated last week
- GPU-Accelerated Radial Basis Function (RBF) Interpolation in PyTorch☆42Updated 6 months ago
- Repo to the paper "Message Passing Neural PDE Solvers"☆145Updated last year
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆212Updated 5 months ago
- Use scipy.optimize.minimize as a PyTorch Optimizer.☆74Updated last year
- Solving Inverse Physics Problems with Score Matching☆32Updated 2 years ago
- NeuralGCM's differentiable dycore☆314Updated last week
- PDE-Transformer is a neural network architecture designed to efficiently process and predict the evolution of physical systems described …☆64Updated 3 months ago
- [ICLR2025 Spotlight] Official implementation of Conflict-Free Inverse Gradients Method☆100Updated 2 months ago
- ☆219Updated 11 months ago
- Official implementation of Diffusion Graph Networks (DGNs)☆77Updated 10 months ago