sail-sg / stde
Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024
☆99Updated 2 months ago
Alternatives and similar repositories for stde:
Users that are interested in stde are comparing it to the libraries listed below
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆49Updated last year
- ☆127Updated last year
- ICON for in-context operator learning☆45Updated 2 months ago
- Official PyTorch implementation of the Vectorized Conditional Neural Field.☆12Updated 6 months ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆97Updated 10 months ago
- ☆241Updated 5 months ago
- Code for the paper Universal Physics Transformers☆89Updated last week
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆44Updated last week
- Code for "DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training"☆49Updated 8 months ago
- Repo to the paper "Message Passing Neural PDE Solvers"☆127Updated 5 months ago
- Neural Diffusion Processes☆76Updated 6 months ago
- Flow-matching algorithms in JAX☆83Updated 6 months ago
- Modern Fixed Point Systems using Pytorch☆86Updated last year
- Neural Stochastic PDEs: resolution-invariant modelling of continuous spatiotemporal dynamics☆49Updated 2 years ago
- Spectral Neural Operator☆75Updated last year
- Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…☆70Updated 8 months ago
- ☆13Updated last year
- Code for Lie Symmetries SSL paper☆19Updated 11 months ago
- Automatic Functional Differentiation in JAX☆63Updated last month
- Model hub for all your DiffeqML needs. Pretrained weights, modules, and basic inference infrastructure☆24Updated last year
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆144Updated 2 months ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆36Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆54Updated 11 months ago
- Code for Characterizing Scaling and Transfer Learning Behavior of FNO in SciML☆42Updated last year
- Pytorch-like dataloaders for JAX.☆75Updated 4 months ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆59Updated 3 months ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆56Updated 5 months ago
- Official implementation of NeurIPS 2024 paper "DiffusionPDE: Generative PDE-Solving Under Partial Observation"☆104Updated 3 months ago
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆130Updated 2 months ago
- code for "Riemannian Flow Matching on General Geometries".☆200Updated 11 months ago