NVIDIA / torch-harmonics
Differentiable signal processing on the sphere for PyTorch
☆444Updated last week
Alternatives and similar repositories for torch-harmonics:
Users that are interested in torch-harmonics are comparing it to the libraries listed below
- ☆115Updated 3 months ago
- A parallel ODE solver for PyTorch☆251Updated 5 months ago
- ☆250Updated 6 months ago
- Newton and Quasi-Newton optimization with PyTorch☆341Updated last year
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆138Updated 3 months ago
- Code for the paper Universal Physics Transformers☆101Updated last month
- Massively parallel training of machine-learning based weather and climate models☆247Updated 2 months ago
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆247Updated 2 years ago
- Repo to the paper "Message Passing Neural PDE Solvers"☆132Updated 6 months ago
- Differentiable and accelerated spherical transforms with JAX☆146Updated this week
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆317Updated 2 weeks ago
- Official implementation of NeurIPS 2024 paper "DiffusionPDE: Generative PDE-Solving Under Partial Observation"☆117Updated 5 months ago
- code for "Riemannian Flow Matching on General Geometries".☆210Updated last year
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆65Updated 4 months ago
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆83Updated 3 months ago
- ☆153Updated last year
- This repository is the official implementation of the paper Convolutional Neural Operators for robust and accurate learning of PDEs☆161Updated 5 months ago
- Geometry-Aware Fourier Neural Operator (Geo-FNO)☆228Updated last year
- NeuralGCM's differentiable dycore☆258Updated 3 weeks ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆59Updated 2 months ago
- Official implementation of Diffusion Graph Networks (DGNs)☆26Updated 2 weeks ago
- Run PyTorch in JAX. 🤝☆232Updated last month
- Spectral Neural Operator☆76Updated last year
- An Online Deep Learning Interface for HPC programs on NVIDIA GPUs☆164Updated this week
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆102Updated 4 months ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆62Updated 2 months ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆83Updated this week
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆151Updated 3 months ago
- Time- and space-continuous neural PDE forecaster based on INRs and ODEs☆58Updated last year
- Adaptive FNO transformer - official Pytorch implementation☆241Updated 2 years ago