NVIDIA / torch-harmonicsLinks
Differentiable signal processing on the sphere for PyTorch
☆463Updated 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☆255Updated 8 months ago
- S2FFT: Differentiable and accelerated spherical transforms☆192Updated this week
- ☆117Updated 6 months ago
- Massively parallel training of machine-learning based weather and climate models☆250Updated 2 weeks ago
- ☆260Updated 8 months ago
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆251Updated 2 years ago
- Newton and Quasi-Newton optimization with PyTorch☆347Updated last week
- Normalizing flows in PyTorch☆386Updated this week
- Official implementation of NeurIPS 2024 paper "DiffusionPDE: Generative PDE-Solving Under Partial Observation"☆126Updated last month
- code for "Riemannian Flow Matching on General Geometries".☆227Updated last year
- Code for the paper Universal Physics Transformers☆113Updated 3 months ago
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆164Updated 5 months ago
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆144Updated last month
- Repo to the paper "Message Passing Neural PDE Solvers"☆134Updated 8 months ago
- A Python library for end-to-end learning on surfaces. It implements pre-processing functions that include geodesic algorithms, neural net…☆33Updated this week
- GPU-Accelerated Radial Basis Function (RBF) Interpolation in PyTorch☆32Updated 2 years ago
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆59Updated 2 months ago
- DeepSphere: a graph-based spherical CNN (PyTorch)☆159Updated 3 years ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆203Updated 6 months ago
- ☆437Updated 3 weeks ago
- ☆190Updated 3 months ago
- NeuralGCM's differentiable dycore☆272Updated last month
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆86Updated 5 months ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆67Updated 4 months ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆73Updated 2 weeks ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆90Updated 2 weeks ago
- Using graph network to solve PDEs☆392Updated last year
- ☆152Updated 2 years ago
- Fast, Expressive SE(n) Equivariant Networks through Weight-Sharing in Position-Orientation Space.☆82Updated last year
- Interpolation routines in Pytorch.☆110Updated 4 years ago