Drop-in environment replacements that make your RL algorithm train faster.
☆21Jun 19, 2024Updated last year
Alternatives and similar repositories for synthetic-gymnax
Users that are interested in synthetic-gymnax are comparing it to the libraries listed below
Sorting:
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆263Oct 31, 2025Updated 4 months ago
- ☆16Jul 16, 2024Updated last year
- Actor-Sharer-Learner training framework for off-policy DRL algorithms☆22Dec 29, 2024Updated last year
- ☆22Jan 15, 2026Updated last month
- ☆92Feb 16, 2026Updated 3 weeks ago
- POPGym Library in JAX☆12Apr 15, 2024Updated last year
- [ICML 2023] Variational Curriculum Reinforcement Learning for Unsupervised Discovery of Skills☆12Jul 15, 2023Updated 2 years ago
- ☆10Jun 27, 2024Updated last year
- Official Code Repo for the paper "Learning to Play Atari in a World of Tokens" accepted at ICML, 2024☆11Jun 6, 2024Updated last year
- Contains JAX implementation of algorithms for inverse reinforcement learning☆74Aug 18, 2024Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆112Dec 5, 2023Updated 2 years ago
- OpenAI gym environments for goal-conditioned and language-conditioned reinforcement learning☆14Jan 27, 2026Updated last month
- Official repository for "Investigating Pre-Training Objectives for Generalization in Visual Reinforcement Learning" (ICML 2024)☆11Sep 16, 2025Updated 5 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 2 years ago
- Learning from preferences is a common paradigm for fine-tuning language models. Yet, many algorithmic design decisions come into play. Ou…☆32Apr 20, 2024Updated last year
- A Dual-RL method DVL: Dual-V Learning for offline and online reinforcement learning☆15Oct 22, 2023Updated 2 years ago
- WebGym: Web-browser-based tasks for RL Agents☆24Feb 4, 2021Updated 5 years ago
- Official codebase for "Sampling For Learnability", published at NeurIPS 2024☆20Oct 21, 2025Updated 4 months ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆26Jan 14, 2025Updated last year
- Simple single-file baselines for Q-Learning in pure-GPU setting☆237Nov 24, 2025Updated 3 months ago
- MarketGPT: Developing a Pre-trained transformer (GPT) for Modeling Financial Time Series☆17Sep 5, 2025Updated 6 months ago
- Neuroevolution Benchmark in JAX 🦕☆42Nov 5, 2023Updated 2 years ago
- Clean single-file implementation of offline RL algorithms in JAX☆170Nov 24, 2025Updated 3 months ago
- PoE-World: Compositional World Modeling with Products of Programmatic Experts☆39Feb 5, 2026Updated last month
- ☆19Apr 22, 2024Updated last year
- GPT implementation in Flax☆18Jan 8, 2022Updated 4 years ago
- Official code for ICML 2024 paper Reinformer: Max-Return Sequence Modeling for offline RL☆46Oct 16, 2024Updated last year
- Lectures on Quantitative Economics Using JAX☆48Feb 28, 2026Updated last week
- Unified Implementations of Offline Reinforcement Learning Algorithms☆200Dec 19, 2025Updated 2 months ago
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆23Oct 28, 2024Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆90Jun 4, 2024Updated last year
- Scalable Opponent Shaping Experiments in JAX☆25Apr 13, 2024Updated last year
- This repository is the official implementation of the TRAC optimizer in Fast TRAC: A Parameter-Free Optimizer for Lifelong Reinforcement …☆33May 2, 2025Updated 10 months ago
- ☆93Jan 21, 2026Updated last month
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆396Mar 1, 2026Updated last week
- Your favourite classical machine learning algos on the GPU/TPU☆22Dec 14, 2025Updated 2 months ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️☆325Dec 16, 2025Updated 2 months ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Oct 12, 2023Updated 2 years ago