shacklettbp / madrona

Related projects ⓘ

Alternatives and complementary repositories for madrona

• NVlabs / DiffRL
  [ICLR 2022] Accelerated Policy Learning with Parallel Differentiable Simulation
  ☆264Updated 6 months ago
• shacklettbp / madrona_escape_room
  ☆18Updated last month
• Emerge-Lab / gpudrive
  GPU-acceleration of Nocturne via Madrona
  ☆227Updated this week
• facebookresearch / minimax
  Efficient baselines for autocurricula in JAX.
  ☆173Updated 2 months ago
• Avalon-Benchmark / avalon
  A 3D video game environment and benchmark designed from scratch for reinforcement learning research
  ☆181Updated last year
• FLAIROx / JaxMARL
  Multi-Agent Reinforcement Learning with JAX
  ☆438Updated last week
• vwxyzjn / cleanba
  CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL
  ☆105Updated 2 months ago
• pytorch-labs / LeanRL
  LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
  ☆432Updated 2 weeks ago
• inspirai / TimeChamber
  A Massively Parallel Large Scale Self-Play Framework
  ☆315Updated last year
• keraJLi / rejax
  ☆147Updated 2 months ago
• araffin / sbx
  SBX: Stable Baselines Jax (SB3 + Jax)
  ☆337Updated last week
• sail-sg / envpool
  C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.
  ☆1,092Updated 3 months ago
• ikostrikov / rlpd
  ☆223Updated last year
• lucidrains / q-transformer
  Implementation of Q-Transformer, Scalable Offline Reinforcement Learning via Autoregressive Q-Functions, out of Google Deepmind
  ☆341Updated last month
• Farama-Foundation / Shimmy
  An API conversion tool for popular external reinforcement learning environments
  ☆139Updated last month
• mlech26l / gigastep
  ☆66Updated 10 months ago
• andyljones / megastep
  megastep helps you build 1-million FPS reinforcement learning environments on a single GPU
  ☆133Updated 2 years ago
• luchris429 / purejaxrl
  Really Fast End-to-End Jax RL Implementations
  ☆717Updated 2 months ago
• bsarkar321 / madrona_rl_envs
  Collection of RL Environments built using Madrona
  ☆25Updated last year
• mttga / purejaxql
  Simple single-file baselines for Q-Learning in pure-GPU setting
  ☆94Updated 3 months ago
• nicklashansen / tdmpc2
  Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"
  ☆391Updated this week
• lowrollr / turbozero
  fast + parallel AlphaZero in JAX
  ☆84Updated 7 months ago
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆117Updated 3 months ago
• alex-petrenko / megaverse
  High-throughput simulation platform for Artificial Intelligence reseach
  ☆219Updated last year
• openrlbenchmark / openrlbenchmark
  ☆200Updated 9 months ago
• MichaelTMatthews / Craftax
  (Crafter + NetHack) in JAX. ICML 2024 Spotlight.
  ☆203Updated 3 weeks ago
• instadeepai / flashbax
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆212Updated 2 weeks ago
• ikostrikov / walk_in_the_park
  ☆244Updated 2 years ago
• facebookresearch / dcd
  Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.
  ☆126Updated 2 months ago