nmboffi / jax-interpolantsLinks
Definitive implementation of the stochastic interpolant framework for generative modeling in jax.
☆29Updated last month
Alternatives and similar repositories for jax-interpolants
Users that are interested in jax-interpolants are comparing it to the libraries listed below
Sorting:
- Flow-matching algorithms in JAX☆100Updated last year
- ☆24Updated 7 months ago
- ☆115Updated 2 months ago
- ☆38Updated last month
- ☆68Updated last year
- Neural Optimal Transport with Lagrangian Costs☆57Updated 2 months ago
- Comparison between GFlowNets & Maximum Entropy RL☆19Updated last year
- High quality implementations of imitation and inverse reinforcement learning algorithms☆19Updated last month
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆97Updated last week
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆130Updated 3 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆96Updated this week
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆27Updated last year
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆28Updated 2 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆35Updated this week
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆32Updated last year
- ☆27Updated last year
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆114Updated 3 weeks ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆78Updated 6 months ago
- ☆35Updated 3 years ago
- Implementation of Computer Vision Models in JAX (equinox)☆12Updated this week
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆110Updated last year
- Unified Implementations of Offline Reinforcement Learning Algorithms☆88Updated 3 months ago
- Pytorch-like dataloaders for JAX.☆94Updated 2 months ago
- Official Code for "Relative Entropy Pathwise Policy Optimization"☆23Updated last week
- The official implementation of flow Q-learning (FQL)☆207Updated 3 weeks ago
- Official code for "World Models via Policy-Guided Trajectory Diffusion", TMLR 2024☆66Updated last year
- iQRL: implicitly Quantized Representations for Sample-efficient Reinforcement Learning☆12Updated 7 months ago
- ☆45Updated 2 months ago
- Building blocks for productive research☆59Updated last week
- A dataloader, but for JAX☆20Updated last year