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
  ☆104Updated last year
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆57Updated 3 months ago
• sukhijab / maxinforl_jax
  ☆24Updated 8 months ago
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆19Updated last year
• kvfrans / splus
  ☆115Updated 2 months ago
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆20Updated last year
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆97Updated last month
• antonioterpin / jkonet-star
  JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".
  ☆31Updated 5 months ago
• vladisai / PLDM
  ☆40Updated last month
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆99Updated 3 weeks ago
• ran-weii / cleanil
  High quality implementations of imitation and inverse reinforcement learning algorithms
  ☆19Updated 2 weeks ago
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆118Updated last month
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆78Updated 10 months ago
• fabian-sp / MoMo
  MoMo: Momentum Models for Adaptive Learning Rates
  ☆19Updated last year
• NanboLi / FACTS
  [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"
  ☆28Updated 3 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆34Updated last year
• yilundu / ired_code_release
  ☆69Updated last year
• 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
  ☆130Updated 4 months ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆38Updated 2 weeks ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆94Updated 3 months ago
• clementpoiret / Equimo
  Implementation of Computer Vision Models in JAX (equinox)
  ☆13Updated last week
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆110Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆42Updated 8 months ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆37Updated last year
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆35Updated 3 years ago
• cvoelcker / reppo
  Official Code for "Relative Entropy Pathwise Policy Optimization"
  ☆24Updated last week
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆82Updated 7 months ago
• seohongpark / fql
  The official implementation of flow Q-learning (FQL)
  ☆214Updated last month
• nikihowe / myriad
  Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.
  ☆67Updated last year
• aidanscannell / iqrl
  iQRL: implicitly Quantized Representations for Sample-efficient Reinforcement Learning
  ☆12Updated 7 months ago