yardenas / jax-dreamer

Related projects: ⓘ

• danijar / crafter-baselines
  Docker containers of baseline agents for the Crafter environment
  ☆27Updated 2 years ago
• MichalBortkiewicz / JaxGCRL
  ☆25Updated last week
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆39Updated 2 years ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 2 years ago
• Howuhh / sac-n-jax
  Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch
  ☆46Updated last year
• danijar / ninjax
  General Modules for JAX
  ☆57Updated last month
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆52Updated 2 years ago
• bmazoure / ppo_jax
  Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…
  ☆49Updated 2 years ago
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆52Updated last year
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• google-deepmind / csuite
  ☆41Updated 5 months ago
• MichaelTMatthews / Craftax_Baselines
  ☆14Updated 2 months ago
• eloialonso / iris_pretrained_models
  ☆11Updated this week
• chamorajg / pl-dreamer
  Simplistic Pytorch Implementation of the Dreamer-RL
  ☆20Updated 2 years ago
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆17Updated last month
• DramaCow / jaxued
  ☆56Updated last month
• Michael-Beukman / RobocupGym
  Reinforcement Learning inside a 3D soccer simulation
  ☆19Updated this week
• jannerm / gamma-models
  Code for the paper "Gamma-Models: Generative Temporal Difference Learning for Infinite-Horizon Prediction"
  ☆39Updated last year
• ElisevanderPol / symmetrizer
  ☆27Updated 3 years ago
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 2 years ago
• keraJLi / synthetic-gymnax
  ☆17Updated 3 months ago
• joeybose / FloRL
  Implicit Normalizing Flows + Reinforcement Learning
  ☆60Updated 5 years ago
• xingchenwan / bgpbt
  [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)
  ☆26Updated 2 years ago
• facebookresearch / svg
  On the model-based stochastic value gradient for continuous reinforcement learning
  ☆54Updated last year
• RobertTLange / gymnax-blines
  Baselines for gymnax 🤖
  ☆57Updated last year
• seohongpark / METRA
  METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)
  ☆49Updated 11 months ago
• tinker495 / jax-baseline
  Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…
  ☆33Updated last week
• hamishs / JAX-RL
  JAX implementations of various deep reinforcement learning algorithms.
  ☆18Updated 3 years ago
• kvfrans / fre
  Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"
  ☆51Updated 5 months ago
• denisyarats / proto
  Proto-RL: Reinforcement Learning with Prototypical Representations
  ☆81Updated 2 years ago