sail-sg / stde
Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024
☆104Updated 5 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
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆100Updated last year
- ☆142Updated last year
- Official PyTorch implementation of the Vectorized Conditional Neural Field.☆13Updated 9 months ago
- Codomain attention neural operator for single to multi-physics PDE adaptation.☆59Updated last month
- Automatic Functional Differentiation in JAX☆72Updated last month
- Flow-matching algorithms in JAX☆90Updated 8 months ago
- Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.☆77Updated 3 weeks ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆38Updated 2 years ago
- ICON for in-context operator learning☆51Updated 2 months ago
- Modern Fixed Point Systems using Pytorch☆89Updated last year
- Code for "DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training"☆55Updated 11 months ago
- code for "Riemannian Flow Matching on General Geometries".☆224Updated last year
- Neural Stochastic PDEs: resolution-invariant modelling of continuous spatiotemporal dynamics☆52Updated 2 years ago
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆22Updated 2 months ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated last month
- Spectral Neural Operator☆78Updated last year
- Collecting research materials on neural samplers with diffusion/flow models☆46Updated 2 months ago
- Official implementation of NeurIPS 2024 paper "DiffusionPDE: Generative PDE-Solving Under Partial Observation"☆124Updated last week
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆57Updated last year
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆32Updated 5 months ago
- Neural Diffusion Processes☆77Updated 9 months ago
- Code release for "Stochastic Optimal Control Matching"☆34Updated 8 months ago
- ☆13Updated last year
- ☆27Updated last year
- Code for the book "The Elements of Differentiable Programming".☆84Updated last month
- ☆256Updated 7 months ago
- Time- and space-continuous neural PDE forecaster based on INRs and ODEs☆64Updated last year
- Code for Characterizing Scaling and Transfer Learning Behavior of FNO in SciML☆44Updated last year
- Code for the paper Universal Physics Transformers☆111Updated 2 months ago