Alternatives and similar repositories for jax-interpolants

Users that are interested in jax-interpolants are comparing it to the libraries listed below

• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆105Updated last year
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆58Updated 4 months ago
• kvfrans / splus
  ☆120Updated 3 months ago
• yilundu / ired_code_release
  ☆71Updated last year
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆121Updated 2 months ago
• ran-weii / cleanil
  High quality implementations of imitation and inverse reinforcement learning algorithms
  ☆20Updated last month
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆102Updated last month
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆19Updated last year
• NanboLi / FACTS
  [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"
  ☆28Updated 4 months ago
• sukhijab / maxinforl_jax
  ☆25Updated 9 months ago
• vladisai / PLDM
  ☆41Updated 2 months ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆27Updated last year
• lollcat / fab-torch
  Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.
  ☆65Updated last year
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆38Updated last month
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆99Updated last month
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆37Updated last year
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆35Updated 3 years ago
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆20Updated last year
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆83Updated 11 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆42Updated 9 months ago
• GFNOrg / gfn-diffusion
  ☆33Updated 6 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆37Updated last year
• DenisBless / variational_sampling_methods
  [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".
  ☆37Updated 9 months ago
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆111Updated 5 months ago
• enjeeneer / zero-shot-rl
  VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)
  ☆20Updated 8 months ago
• EmptyJackson / policy-guided-diffusion
  Official implementation of the RLC 2024 paper "Policy-Guided Diffusion"
  ☆142Updated last year
• FlyingGiraffe / dhn
  Denoising Hamiltonian Network for Physical Reasoning
  ☆37Updated 6 months ago
• lucidrains / improving-transformers-world-model-for-rl
  Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch
  ☆136Updated 5 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆94Updated 4 months ago
• clementpoiret / Equimo
  Implementation of Computer Vision Models in JAX (equinox)
  ☆13Updated 2 weeks ago