Implementation of Proximal Policy Optimization in Jax+Flax
☆21May 18, 2023Updated 3 years ago
Alternatives and similar repositories for Recurrent-PPO-Jax
Users that are interested in Recurrent-PPO-Jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.☆21Feb 20, 2023Updated 3 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 11 months ago
- PyTorch-ready drop-in Triton-based Gated SSM for ultra-long sequences. Heavily optimized. Suitable for pre-training and fine-tuning.☆10Dec 16, 2025Updated 10 months ago
- ☆10Jun 27, 2024Updated 2 years 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.
- ☆15Jul 16, 2021Updated 5 years ago
- 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
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 3 years ago
- Official implementation for "How Should We Meta-Learn Reinforcement Learning Algorithms?"☆23Sep 7, 2025Updated last year
- Decision Transformer JAX - Reproduction of 'Decision Transformer: Reinforcement Learning via Sequence Modeling' in JAX and Haiku☆13Aug 14, 2024Updated 2 years ago
- Partially Observable Benchmarks in JAX☆29Apr 30, 2026Updated 6 months ago
- Reproduction of Curiosity-driven Exploration by Self-supervised Prediction in PyTorch☆12Jun 10, 2019Updated 7 years ago
- Toy meta-RL environments for testing algorithms implementations☆10Feb 21, 2024Updated 2 years ago
- ☆16Sep 28, 2022Updated 4 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- A DDPM implementation in Jax for continuous space modeling.☆10Apr 24, 2023Updated 3 years ago
- Official implementation for "Let Offline RL Flow: Training Conservative Agents in the Latent Space of Normalizing Flows", NeurIPS 2022, O…☆12Jan 31, 2023Updated 3 years ago
- ☆18Apr 22, 2024Updated 2 years ago
- Sample-Efficient Online Learning in LM Agents via Hindsight Trajectory Rewriting☆14Jan 2, 2026Updated 10 months ago
- Code for LLM_Catastrophic_Forgetting via SAM.☆11Jun 7, 2024Updated 2 years ago
- Typescript implementation of Kinetix☆12Feb 25, 2025Updated last year
- ☆36Nov 22, 2024Updated last year
- Official implementation of the paper "Arbitrarily Scalable Environment Generators via Neural Cellular Automata", published in NeurIPS 202…☆13Oct 31, 2023Updated 3 years ago
- Implementation of our paper "Data Rejuvenation: Exploiting Inactive Training Examples for Neural Machine Translation" in EMNLP-2020.☆23Aug 20, 2021Updated 5 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- ☆11Nov 13, 2024Updated last year
- AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)☆24Oct 15, 2024Updated 2 years ago
- Official implementation for "Anti-Exploration by Random Network Distillation", ICML 2023☆58Feb 3, 2023Updated 3 years ago
- ☆16Jul 16, 2024Updated 2 years ago
- ☆49Jul 22, 2026Updated 3 months ago
- A simple demonstration of how to train an LSTM model with Reinforcement Learning using PyTorch☆12Jun 15, 2026Updated 4 months ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆25Feb 15, 2025Updated last year
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆14Jan 3, 2023Updated 3 years ago
- ☆98Feb 16, 2026Updated 8 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Baseline implementation of recurrent PPO using truncated BPTT☆160Apr 28, 2024Updated 2 years ago
- An official implementation for the EMNLP 2023 Findings paper "Prompt-Based Editing for Text Style Transfer"☆13Dec 9, 2023Updated 2 years ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆29Oct 29, 2023Updated 3 years ago
- Model-Based Uncertainty in Value Functions (AISTATS2023)☆16Feb 28, 2023Updated 3 years ago
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆13Aug 31, 2020Updated 6 years ago
- ☆18Jun 2, 2026Updated 5 months ago
- ☆50Nov 20, 2025Updated 11 months ago