Alternatives and similar repositories for jax-interpolants

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

• nmboffi / flow-maps
  Official codebase for the paper "How to build a consistency model: Learning flow maps via self-distillation" (NeurIPS 2025).
  ☆48Updated last month
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆107Updated last year
• willisma / diffuse_nnx
  A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…
  ☆116Updated last month
• sukhijab / maxinforl_jax
  ☆26Updated last month
• kvfrans / splus
  ☆119Updated 5 months ago
• vladisai / PLDM
  ☆47Updated 3 weeks ago
• yilundu / ired_code_release
  ☆73Updated last year
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆120Updated last month
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆19Updated last year
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆60Updated 6 months ago
• klieret / wandb-offline-sync-hook
  A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!
  ☆86Updated 2 weeks ago
• ran-weii / cleanil
  High quality implementations of imitation and inverse reinforcement learning algorithms
  ☆21Updated 3 months ago
• seohongpark / fql
  The official implementation of flow Q-learning (FQL)
  ☆251Updated 4 months ago
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆92Updated 10 months ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆28Updated last year
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆36Updated 4 years ago
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆125Updated 3 months ago
• NanboLi / FACTS
  [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"
  ☆28Updated 5 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆39Updated last year
• kvfrans / cfgrl
  ☆57Updated 5 months ago
• google-deepmind / dmc_vision_benchmark
  ☆28Updated last year
• kvfrans / lmpo
  ☆122Updated this week
• EmptyJackson / policy-guided-diffusion
  Official implementation of the RLC 2024 paper "Policy-Guided Diffusion"
  ☆148Updated last year
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆105Updated last month
• 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
  ☆144Updated 6 months ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆38Updated last month
• 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
• masa-ue / RLfinetuning_Diffusion_Bioseq
  Code for the tutorial/review paper for RL-based-fine-tuniing. In this code, we especially focus on the design of biological sequences li…
  ☆151Updated last year
• cvoelcker / reppo
  Official Code for "Relative Entropy Pathwise Policy Optimization"
  ☆39Updated this week
• aidanscannell / iqrl
  iQRL: implicitly Quantized Representations for Sample-efficient Reinforcement Learning
  ☆12Updated 10 months ago