sail-sg / stdeLinks
Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024
☆108Updated 6 months ago
Alternatives and similar repositories for stde
Users that are interested in stde are comparing it to the libraries listed below
Sorting:
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆50Updated 2 years ago
- ☆146Updated last year
- Modern Fixed Point Systems using Pytorch☆93Updated last year
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆101Updated last year
- Flow-matching algorithms in JAX☆92Updated 9 months ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆77Updated last month
- Collecting research materials on neural samplers with diffusion/flow models☆48Updated last week
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆24Updated this week
- code for "Riemannian Flow Matching on General Geometries".☆227Updated last year
- Automatic Functional Differentiation in JAX☆72Updated 2 months ago
- Neural Stochastic PDEs: resolution-invariant modelling of continuous spatiotemporal dynamics☆52Updated 2 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆62Updated 2 months ago
- Official implementation of NeurIPS 2024 paper "DiffusionPDE: Generative PDE-Solving Under Partial Observation"☆126Updated last month
- ☆260Updated 8 months ago
- ☆27Updated last year
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆38Updated 2 years ago
- Official PyTorch implementation of the Vectorized Conditional Neural Field.☆14Updated 10 months ago
- ICON for in-context operator learning☆53Updated 3 months ago
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆59Updated 2 months ago
- Code for the book "The Elements of Differentiable Programming".☆86Updated 2 months ago
- Repo to the paper "Message Passing Neural PDE Solvers"☆134Updated 8 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆58Updated last year
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆164Updated 5 months ago
- ☆13Updated last year
- ☆16Updated last year
- Code for Lie Symmetries SSL paper☆22Updated last year
- Code for the paper "Poseidon: Efficient Foundation Models for PDEs"☆144Updated last month
- Stochastic Normalizing Flows☆76Updated 3 years ago
- Lightning-like training API for JAX with Flax☆38Updated 5 months ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆48Updated 3 weeks ago