Alternatives and similar repositories for jax-dreamer:

Users that are interested in jax-dreamer are comparing it to the libraries listed below

• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆28Updated 3 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• Howuhh / sac-n-jax
  Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch
  ☆47Updated last year
• bmazoure / ppo_jax
  Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…
  ☆53Updated 2 years ago
• danijar / ninjax
  General Modules for JAX
  ☆62Updated 5 months ago
• FLAIROx / popjym
  POPGym Library in JAX
  ☆11Updated 9 months ago
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆54Updated 2 years ago
• chamorajg / pl-dreamer
  Simplistic Pytorch Implementation of the Dreamer-RL
  ☆20Updated 2 years ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆9Updated last year
• DramaCow / jaxued
  ☆66Updated 4 months ago
• FLAIROx / JaxGL
  Simple JAX Graphics Library.
  ☆29Updated 2 months ago
• keraJLi / synthetic-gymnax
  ☆19Updated 6 months ago
• facebookresearch / svg
  On the model-based stochastic value gradient for continuous reinforcement learning
  ☆55Updated last year
• ElisevanderPol / symmetrizer
  ☆29Updated 3 years ago
• brownirl / lambda_discrepancy
  Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy
  ☆17Updated 2 months ago
• RobertTLange / gymnax-blines
  Baselines for gymnax 🤖
  ☆61Updated last year
• google-deepmind / csuite
  ☆43Updated 3 months ago
• FLAIROx / jaxirl
  Contains JAX implementation of algorithms for inverse reinforcement learning
  ☆67Updated 5 months ago
• MichaelTMatthews / Craftax_Baselines
  ☆18Updated last week
• eilab-gt / NovGrid
  Novelty MiniGrid--NovGrid--is an extension of MiniGrid environment that allows for the world properties and dynamics to change according …
  ☆35Updated 7 months ago
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆36Updated 5 months ago
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆79Updated 2 years ago
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated last year
• symoon11 / dreamerv3-flax
  Flax Implementation of DreamerV3 on Crafter
  ☆10Updated 10 months ago
• kenjyoung / dreamerv2_JAX
  An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.
  ☆15Updated 2 years ago
• MarcoMeter / neroRL
  Deep Reinforcement Learning Framework done with PyTorch
  ☆31Updated last week
• flowersteam / vivarium
  Multi-agent simulator in Jax for research and teaching in AI & ALife
  ☆15Updated this week
• qgallouedec / lge
  ☆31Updated 9 months ago