Alternatives and similar repositories for stde

Users that are interested in stde are comparing it to the libraries listed below

• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆52Updated 2 years ago
• crispitagorico / torchspde
  Neural Stochastic PDEs: resolution-invariant modelling of continuous spatiotemporal dynamics
  ☆53Updated 2 years ago
• malbergo / stochastic-interpolants
  ☆167Updated last year
• alisiahkoohi / Langevin-dynamics
  Sampling with gradient-based Markov Chain Monte Carlo approaches
  ☆107Updated last year
• locuslab / torchdeq
  Modern Fixed Point Systems using Pytorch
  ☆103Updated last year
• PolymathicAI / multiple_physics_pretraining
  Code for paper "Multiple Physics Pretraining for Physical Surrogate Models
  ☆169Updated 8 months ago
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆65Updated 4 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆100Updated 11 months ago
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆74Updated 4 months ago
• facebookresearch / neural-conservation-law
  code for "Neural Conservation Laws A Divergence-Free Perspective".
  ☆39Updated 2 years ago
• ml-jku / UPT
  Code for the paper Universal Physics Transformers
  ☆124Updated last month
• neuraloperator / DiffusionOperators
  ☆13Updated last year
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆58Updated last year
• J-zin / Awesome-Diffusion-Flow-Samplers
  Collecting research materials on neural samplers with diffusion/flow models
  ☆58Updated last month
• necludov / jam
  Implementation of Action Matching
  ☆44Updated 2 years ago
• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
  ☆99Updated last month
• martenlienen / torchode
  A parallel ODE solver for PyTorch
  ☆263Updated 10 months ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆36Updated 11 months ago
• pdearena / pdearena
  ☆267Updated 10 months ago
• sony / fp-diffusion
  ☆28Updated last year
• jhagnberger / vcnef
  Official PyTorch implementation of the Vectorized Conditional Neural Field.
  ☆15Updated last year
• DoronHav / WassersteinFlowMatching
  ☆29Updated this week
• YWolfeee / lapjax
  A JAX based package designed for efficient second order operators (e.g., laplacian) computation.
  ☆72Updated last year
• facebookresearch / riemannian-fm
  code for "Riemannian Flow Matching on General Geometries".
  ☆247Updated last year
• samholt / NeuralLaplace
  Neural Laplace: Differentiable Laplace Reconstructions for modelling any time observation with O(1) complexity.
  ☆80Updated 3 months ago
• scaomath / galerkin-transformer
  [NeurIPS 2021] Galerkin Transformer: a linear attention without softmax for Partial Differential Equations
  ☆248Updated last year
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆114Updated 2 weeks ago
• neuraloperator / CoDA-NO
  Codomain attention neural operator for single to multi-physics PDE adaptation.
  ☆63Updated 2 months ago
• brandstetter-johannes / MP-Neural-PDE-Solvers
  Repo to the paper "Message Passing Neural PDE Solvers"
  ☆135Updated 11 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆42Updated 8 months ago