Alternatives and similar repositories for lagrangian-ot:

Users that are interested in lagrangian-ot are comparing it to the libraries listed below

• maxxxzdn / clifford-group-equivariant-cnns
  Code repository of the paper "Clifford-Steerable Convolutional Neural Networks"
  ☆46Updated 6 months ago
• DoronHav / WassersteinFlowMatching
  ☆21Updated 2 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆83Updated 6 months ago
• lazaratan / meta-flow-matching
  Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
  ☆45Updated 5 months ago
• TianrongChen / DMSB
  Official implementation of Deep Momentum Schrödinger Bridge
  ☆23Updated last year
• vislearn / FFF
  Free-form flows are a generative model training a pair of neural networks via maximum likelihood
  ☆38Updated 3 weeks ago
• oxcsml / constrained-diffusion
  ☆16Updated last year
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆30Updated 6 months ago
• necludov / wl-mechanics
  ☆32Updated 8 months ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆43Updated last year
• ghliu / DeepGSB
  Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral
  ☆47Updated 2 years ago
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆34Updated 3 years ago
• sophiaas / spectral-universality
  Official PyTorch and JAX Implementation of "Harmonics of Learning: Universal Fourier Features Emerge in Invariant Networks"
  ☆36Updated last year
• facebookresearch / riemannian-fm
  code for "Riemannian Flow Matching on General Geometries".
  ☆200Updated 11 months ago
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆55Updated 8 months ago
• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆78Updated 10 months ago
• antonioterpin / jkonet-star
  JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".
  ☆29Updated last month
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆56Updated last year
• iamalexkorotin / Wasserstein2Benchmark
  A set of tests for evaluating large-scale algorithms for Wasserstein-2 transport maps computation (NeurIPS 2021)
  ☆38Updated 2 years ago
• kksniak / metric-flow-matching
  Official implementation of Metric Flow Matching (NeurIPS 2024)
  ☆30Updated 3 months ago
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆44Updated last year
• yilundu / ired_code_release
  ☆51Updated 8 months ago
• google-deepmind / md4
  Code for https://arxiv.org/abs/2406.04329
  ☆58Updated 2 months ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆45Updated 2 weeks ago
• AllanYangZhou / nfn
  NF-Layers for constructing neural functionals.
  ☆82Updated last year
• mkofinas / neural-graphs
  Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).
  ☆77Updated 6 months ago
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆52Updated 3 months ago
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆24Updated last year
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆17Updated last year
• facebookresearch / SSLForPDEs
  Code for Lie Symmetries SSL paper
  ☆19Updated 11 months ago