NVIDIA / torch-harmonics
Differentiable signal processing on the sphere for PyTorch
☆401Updated last month
Related projects ⓘ
Alternatives and complementary repositories for torch-harmonics
- ☆109Updated 3 months ago
- ☆233Updated 2 months ago
- A parallel ODE solver for PyTorch☆233Updated last month
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆104Updated this week
- Newton and Quasi-Newton optimization with PyTorch☆321Updated 8 months ago
- Numerical integration in arbitrary dimensions on the GPU using PyTorch / TF / JAX☆190Updated 4 months ago
- Official implementation of NeurIPS 2024 paper "DiffusionPDE: Generative PDE-Solving Under Partial Observation"☆77Updated 3 weeks ago
- Code for the paper Universal Physics Transformers☆78Updated last month
- Massively parallel training of machine-learning based weather and climate models☆234Updated last month
- Interpolating natural cubic splines. Includes batching, GPU support, support for missing values, evaluating derivatives of the spline, an…☆228Updated 2 years ago
- Repo to the paper "Message Passing Neural PDE Solvers"☆126Updated 2 months ago
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆126Updated 6 months ago
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆236Updated 2 weeks ago
- Differentiable and accelerated spherical transforms with JAX☆137Updated this week
- Adaptive FNO transformer - official Pytorch implementation☆222Updated 2 years ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆75Updated 2 weeks ago
- ☆215Updated 2 weeks ago
- Spectral Neural Operator☆73Updated 11 months ago
- [ICLR 2023] Factorized Fourier Neural Operators☆149Updated last year
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆54Updated 2 months ago
- Normalizing flows in PyTorch☆330Updated last month
- Geometry-Aware Fourier Neural Operator (Geo-FNO)☆184Updated last year
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulation☆71Updated last month
- ☆58Updated 9 months ago
- This repository is the official implementation of the paper Convolutional Neural Operators for robust and accurate learning of PDEs☆134Updated last month
- ☆145Updated 9 months ago
- code for "Riemannian Flow Matching on General Geometries".☆177Updated 8 months ago
- ☆149Updated 2 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆333Updated this week
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆58Updated 4 months ago