NVIDIA / torch-harmonicsLinks
Differentiable signal processing on the sphere for PyTorch
☆546Updated 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☆270Updated last year
- ☆273Updated last year
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆270Updated 3 years ago
- ☆128Updated 10 months ago
- Code for the paper Universal Physics Transformers☆141Updated 4 months ago
- Newton and Quasi-Newton optimization with PyTorch☆368Updated 5 months ago
- Massively parallel training of machine-learning based weather and climate models☆330Updated 2 weeks ago
- S2FFT: Differentiable and accelerated spherical transforms☆208Updated this week
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆155Updated 6 months ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆83Updated 5 months ago
- Official implementation of NeurIPS 2024 paper "DiffusionPDE: Generative PDE-Solving Under Partial Observation"☆149Updated 6 months ago
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆177Updated 10 months ago
- Normalizing flows in PyTorch☆411Updated 2 weeks ago
- Official implementation of "Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulation"☆43Updated 3 weeks ago
- OSS library that implements deep learning methods for partial differential equations and much more☆454Updated last month
- PyTorch implementation of Levenberg-Marquardt training algorithm☆84Updated 7 months ago
- Repo to the paper "Message Passing Neural PDE Solvers"☆139Updated last year
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆49Updated 5 months ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆92Updated last week
- DeepSphere: a graph-based spherical CNN (PyTorch)☆162Updated 3 years ago
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆69Updated 4 months ago
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆99Updated 10 months ago
- Time- and space-continuous neural PDE forecaster based on INRs and ODEs - ICLR 2023☆74Updated last year
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆75Updated this week
- Official implementation of Diffusion Graph Networks (DGNs)☆70Updated 7 months ago
- [ICLR 2023] Factorized Fourier Neural Operators☆176Updated 2 years ago
- ☆166Updated last year
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆209Updated 2 months ago
- [ICLR2025 Spotlight] Official implementation of Conflict-Free Inverse Gradients Method☆84Updated 5 months ago
- A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs☆184Updated last year