A high-performance reinforcement learning library in jax specialized for robotic learning
☆22Sep 4, 2023Updated 3 years ago
Alternatives and similar repositories for robojax
Users that are interested in robojax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 11 months ago
- Accelerated replay buffers in JAX☆47Sep 17, 2022Updated 4 years ago
- Robotics in Python☆13Feb 21, 2023Updated 3 years ago
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆76Aug 31, 2024Updated 2 years ago
- A modular implementation of PPO, and soon hopefully other algorithms.☆27Jan 16, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Learning Robust Dynamics Through Variational Sparse Gating☆21Oct 19, 2022Updated 4 years ago
- Scripts to recreate the D4RL datasets with Minari☆25Oct 24, 2026Updated last week
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆14Jan 3, 2023Updated 3 years ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆25Feb 15, 2025Updated last year
- Gym env for Slay the Spire☆19Dec 31, 2024Updated last year
- Baselines for gymnax 🤖☆82Apr 3, 2023Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 6 years ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 years ago
- ☆50Nov 20, 2025Updated 11 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- a modular reinforcement learning library with JAX agents☆27Mar 3, 2025Updated last year
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆614Oct 27, 2026Updated last week
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆96Jun 4, 2024Updated 2 years ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆62Aug 22, 2026Updated 2 months ago
- An Agent-Environment interface with Jax transformation compatibility☆13Nov 17, 2023Updated 2 years ago
- A simple sbi tutorial☆13Jul 6, 2025Updated last year
- A Reinforcement Learning Software Toolbox for Robotics☆54Mar 13, 2023Updated 3 years ago
- speed-running solving robot manipulation tasks☆24Oct 31, 2024Updated 2 years ago
- Generator of binary installers for libraries built by vcpkg, alpha version.☆11Aug 5, 2021Updated 5 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- A toolkit for practical Human-AI cooperation research☆15Apr 19, 2024Updated 2 years ago
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 3 years ago
- Collection of robot models compatible with gym-ignition☆26Oct 18, 2021Updated 5 years ago
- ☆19Mar 1, 2023Updated 3 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆107May 17, 2022Updated 4 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
- On the model-based stochastic value gradient for continuous reinforcement learning☆60Mar 6, 2026Updated 7 months ago
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated 2 years ago
- Trajectory-based Deep Generative Replay☆12Jun 19, 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.
- Contains SW specific to the R1 robots☆10Jul 30, 2026Updated 3 months ago
- Simulink toolbox to rapidly prototype robot controllers☆26Jul 17, 2026Updated 3 months ago
- ROS2 / Ignition Gazebo Gym Environment for Reproducible Reinforcement Learning☆13Nov 4, 2021Updated 4 years ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆282Aug 27, 2026Updated 2 months ago
- RL Environments in JAX 🌍☆938Updated this week
- The ultimate Python tool for RObot Descriptions processing.☆18Mar 30, 2026Updated 7 months ago
- Mutual Information State Intrinsic Control (ICLR 2021 Spotlight)☆39Mar 1, 2021Updated 5 years ago