nZhangx / TrajectoryFlowMatching

Related projects ⓘ

Alternatives and complementary repositories for TrajectoryFlowMatching

• necludov / wl-mechanics
  ☆30Updated 5 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆46Updated 3 months ago
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆52Updated 2 months ago
• jmtomczak / vae_kan_example
  A simple example of VAEs with KANs
  ☆12Updated 5 months ago
• opallab / positional_attention
  Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"
  ☆14Updated last month
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆74Updated 3 months ago
• lazaratan / meta-flow-matching
  Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
  ☆39Updated 2 months ago
• gle-bellier / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆51Updated 2 months ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆27Updated 2 months ago
• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆38Updated 9 months ago
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆70Updated 5 months ago
• MaximeRobeyns / bayesian_flow_networks
  A PyTorch implementation of Bayesian flow networks (Graves et al., 2023).
  ☆23Updated 11 months ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆47Updated 3 months ago
• stepelu / idbm-pytorch
  ☆11Updated last year
• GFNOrg / gfn-diffusion
  ☆19Updated last month
• wiseodd / lapeft-bayesopt
  Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.
  ☆15Updated 3 months ago
• kksniak / metric-flow-matching
  Official implementation of Metric Flow Matching (NeurIPS 2024)
  ☆15Updated 2 weeks ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆86Updated 3 weeks ago
• lucidrains / sinkhorn-router-pytorch
  Self contained pytorch implementation of a sinkhorn based router, for mixture of experts or otherwise
  ☆31Updated 2 months ago
• francois-rozet / papers-101
  Implementation of papers in 101 lines of code.
  ☆18Updated last year
• malbergo / stochastic-interpolants
  ☆104Updated 9 months ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆123Updated 10 months ago
• facebookresearch / generalized-schrodinger-bridge-matching
  code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).
  ☆39Updated 7 months ago
• nnaisense / bayesian-flow-networks
  This is the official code release for Bayesian Flow Networks.
  ☆254Updated 3 months ago
• shikaiqiu / compute-better-spent
  ☆46Updated last month
• DenisBless / variational_sampling_methods
  [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".
  ☆25Updated 2 months ago
• mkofinas / neural-graphs
  Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).
  ☆70Updated 3 months ago
• HannesStark / dirichlet-flow-matching
  ☆86Updated 6 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 4 months ago
• apple / ml-agm
  ☆43Updated 6 months ago