Alternatives and similar repositories for jax-interpolants

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

• vladisai / PLDM
  ☆36Updated this week
• sukhijab / maxinforl_jax
  ☆23Updated 7 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆97Updated 11 months ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆56Updated last month
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆108Updated last month
• kvfrans / cfgrl
  ☆43Updated last month
• kvfrans / splus
  ☆111Updated last month
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆107Updated last year
• aidanscannell / iqrl
  iQRL: implicitly Quantized Representations for Sample-efficient Reinforcement Learning
  ☆12Updated 6 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
• NanboLi / FACTS
  [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"
  ☆28Updated last month
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆18Updated last year
• FLAIROx / jafar
  JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"
  ☆67Updated 5 months ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆36Updated 11 months ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆95Updated last month
• seohongpark / fql
  The official implementation of flow Q-learning (FQL)
  ☆177Updated 4 months ago
• google-deepmind / dmc_vision_benchmark
  ☆26Updated last year
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆85Updated 2 months ago
• yilundu / ired_code_release
  ☆67Updated last year
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆30Updated last year
• danijar / elements
  Building blocks for productive research
  ☆59Updated 5 months ago
• ran-weii / cleanil
  High quality implementations of imitation and inverse reinforcement learning algorithms
  ☆19Updated 3 weeks ago
• seohongpark / horizon-reduction
  The official implementation of "Horizon Reduction Makes RL Scalable"
  ☆117Updated last month
• sukhijab / maxinforl_torch
  ☆44Updated 7 months ago
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆97Updated last week
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆48Updated last year
• enjeeneer / zero-shot-rl
  VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)
  ☆17Updated 6 months ago
• aidanscannell / dcmpc
  Official PyTorch implementation of "DC-MPC: Discrete Codebook Model Predictive Control"
  ☆20Updated 4 months ago
• StoneT2000 / robojax
  A high-performance reinforcement learning library in jax specialized for robotic learning
  ☆22Updated last year
• marc-rigter / polygrad-world-models
  Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024
  ☆66Updated last year