A high-performance reinforcement learning library in jax specialized for robotic learning
☆22Sep 4, 2023Updated 2 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☆18Nov 29, 2025Updated 7 months ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Robotics in Python☆13Feb 21, 2023Updated 3 years ago
- Scripts to recreate the D4RL datasets with Minari☆26Jul 4, 2026Updated 3 weeks ago
- A modular implementation of PPO, and soon hopefully other algorithms.☆27Jan 16, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Baselines for gymnax 🤖☆78Apr 3, 2023Updated 3 years ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 years ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆14Jan 3, 2023Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- Gym env for Slay the Spire☆19Dec 31, 2024Updated last year
- a modular reinforcement learning library with JAX agents☆27Mar 3, 2025Updated last year
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆74Aug 31, 2024Updated last year
- A Reinforcement Learning Software Toolbox for Robotics☆54Mar 13, 2023Updated 3 years ago
- speed-running solving robot manipulation tasks☆24Oct 31, 2024Updated last year
- 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.
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 3 years ago
- Collection of robot models compatible with gym-ignition☆26Oct 18, 2021Updated 4 years ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆603Updated this week
- Generator of binary installers for libraries built by vcpkg, alpha version.☆11Aug 5, 2021Updated 4 years ago
- A tool for recording RL trajectories.☆123Updated this week
- A toolkit for practical Human-AI cooperation research☆15Apr 19, 2024Updated 2 years ago
- ☆19Mar 1, 2023Updated 3 years ago
- The ultimate Python tool for RObot Descriptions processing.☆18Mar 30, 2026Updated 3 months ago
- Simulink toolbox to rapidly prototype robot controllers☆25Jul 17, 2026Updated last week
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- On the model-based stochastic value gradient for continuous reinforcement learning☆58Mar 6, 2026Updated 4 months 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
- Benchmarking RL generalization in an interpretable way.☆183Nov 20, 2025Updated 8 months ago
- gym RL environment in which a mujoco simulation of Agility Robotics' Cassie robot is rewarded for walking/running forward as fast as poss…☆35Nov 17, 2023Updated 2 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆95Jun 4, 2024Updated 2 years ago
- A CLI tool for generating Boost.Python/pybind11 bindings from C/C++☆16Aug 22, 2020Updated 5 years ago
- Mutual Information State Intrinsic Control (ICLR 2021 Spotlight)☆39Mar 1, 2021Updated 5 years ago
- Implementation of Proximal Policy Optimization in Jax+Flax☆21May 18, 2023Updated 3 years ago
- Einsum-like high-level array sharding API for JAX☆35Jul 16, 2024Updated 2 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.
- ☆46Jul 12, 2024Updated 2 years ago
- Model-based reinforcement learning (generative simulator models and planning agents)☆16Mar 13, 2026Updated 4 months ago
- A simple library for scaling up JAX programs☆148Nov 4, 2025Updated 8 months ago
- RL Environments in JAX 🌍☆910Apr 2, 2026Updated 3 months ago
- [deprecated] Engine Agnostic Gym Environment for Robotics☆17Feb 10, 2022Updated 4 years ago
- PushWorld: A benchmark for manipulation planning with tools and movable obstacles☆95May 5, 2026Updated 2 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆274Jun 10, 2026Updated last month