nmboffi / jax-interpolantsLinks
Definitive implementation of the stochastic interpolant framework for generative modeling in jax.
☆32Updated 3 months ago
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☆106Updated last year
- ☆25Updated 10 months ago
- Neural Optimal Transport with Lagrangian Costs☆59Updated 5 months ago
- ☆42Updated 3 months ago
- ☆120Updated 4 months ago
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆28Updated 4 months ago
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆106Updated last week
- Comparison between GFlowNets & Maximum Entropy RL☆19Updated last year
- Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"☆28Updated last year
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆38Updated last year
- Building blocks for productive research☆62Updated 2 months ago
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆103Updated 3 weeks ago
- ☆73Updated last year
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆35Updated 5 months ago
- High quality implementations of imitation and inverse reinforcement learning algorithms☆21Updated 2 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆104Updated this week
- Unified Implementations of Offline Reinforcement Learning Algorithms☆115Updated 2 weeks ago
- iQRL: implicitly Quantized Representations for Sample-efficient Reinforcement Learning☆12Updated 9 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆295Updated last year
- A convenient way to trigger synchronizations to wandb / Weights & Biases if your compute nodes don't have internet!☆86Updated 3 weeks ago
- Official Code for "Relative Entropy Pathwise Policy Optimization"☆36Updated 2 weeks ago
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆122Updated 3 months ago
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆72Updated 4 months ago
- ☆35Updated 3 years ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆31Updated 7 months ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆20Updated 2 years ago
- ☆12Updated 5 months ago
- ☆28Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆110Updated last year
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆141Updated 5 months ago