A DDPM implementation in Jax for continuous space modeling.
☆10Apr 24, 2023Updated 3 years ago
Alternatives and similar repositories for JaxDDPM
Users that are interested in JaxDDPM are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- [ICLR 2023 Oral] The official implementation of SQL and EQL in "Offline RL with No OOD Actions: In-Sample Learning via Implicit Value Reg…☆46Jul 27, 2023Updated 3 years ago
- Repo for Implicit Diffusion Q-Learning☆127Dec 5, 2023Updated 2 years ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆83May 13, 2024Updated 2 years ago
- ☆11Apr 16, 2026Updated 6 months ago
- stable-baselines with JAX & Haiku☆13Jun 20, 2024Updated 2 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Corax: Core RL in JAX☆41Feb 22, 2024Updated 2 years ago
- Implementation of GAIL and AIRL using chinerrl☆16Jun 21, 2022Updated 4 years ago
- Code to reproduce Supervised Policy Update (ICLR 2019)☆17Dec 8, 2022Updated 3 years ago
- An implementation of effective policy ensemble.☆16Jul 5, 2023Updated 3 years ago
- ☆26Jan 26, 2024Updated 2 years ago
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆18Nov 24, 2025Updated 11 months ago
- Neuroevolution is a Competitive Alternative to Reinforcement Learning for Skill Discovery☆17Apr 2, 2026Updated 7 months ago
- ☆14Sep 13, 2022Updated 4 years ago
- Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimization☆30Dec 1, 2025Updated 11 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Simple Successor Features☆19Jul 15, 2025Updated last year
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆18Jan 16, 2023Updated 3 years ago
- LXM3: XManager launch backend for HPC clusters☆11May 11, 2024Updated 2 years ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆21Mar 22, 2022Updated 4 years ago
- Code for ICLR 2022 paper Rethinking Goal-Conditioned Supervised Learning and Its Connection to Offline RL.☆28Feb 21, 2022Updated 4 years ago
- Code to related to my NIPS 2016 paper☆10Dec 4, 2016Updated 9 years ago
- Example of android app written in Qt/Qml which uses MXNet for plant image recognition.☆10Nov 4, 2017Updated 9 years ago
- [ICML 2022] The official implementation of DWBC in "Discriminator-Weighted Offline Imitation Learning from Suboptimal Demonstrations"☆37Jan 5, 2023Updated 3 years ago
- ☆11Nov 10, 2022Updated 3 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Implementation of Proximal Policy Optimization in Jax+Flax☆21May 18, 2023Updated 3 years ago
- [NeurIPS 24] Incremental Learning of Retrievable Skills For Efficient Continual Task Adaptation☆29Oct 12, 2025Updated last year
- LLM inference with HuggingFace (experimental)☆11Feb 6, 2025Updated last year
- Fast dataset format and loader☆24Oct 15, 2026Updated 3 weeks ago
- ☆11Jun 1, 2022Updated 4 years ago
- Code accompanying https://arxiv.org/abs/2206.01079☆10Oct 28, 2022Updated 4 years ago
- D3PE (Deep Data-Driven Policy Evaluation) aims to evaluation a large set of candidate policies from a fixed dataset to select best ones.☆10Jun 2, 2022Updated 4 years ago
- ☆22Nov 8, 2021Updated 5 years ago
- Official implementation for the paper "Offline Meta RL - Identifiability Challenges and Effective Data Collection Strategies", NeurIPS 20…☆31Nov 23, 2021Updated 4 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- ☆59Jan 20, 2023Updated 3 years ago
- Test-Time Anchoring for Discrete Diffusion Posterior Sampling☆11Feb 4, 2026Updated 9 months ago
- Official implementation for "Anti-Exploration by Random Network Distillation", ICML 2023☆58Feb 3, 2023Updated 3 years ago
- Code for PuzzleJAX, a benchmark for reasoning and learning, that reimplements PuzzleScript, a concise and expressive DSL and game engine …☆32Oct 28, 2026Updated last week
- An implementation of the paper "Solving the Rubik's Cube without Human Knowledge"☆14Dec 9, 2018Updated 7 years ago
- Change-Based Exploration Transfer☆35Apr 24, 2022Updated 4 years ago
- Codebase for ICLR 2023 paper, "SMART: Self-supervised Multi-task pretrAining with contRol Transformers"☆53Jan 26, 2024Updated 2 years ago