entity-neural-network / enn-trainer

Related projects: ⓘ

• entity-neural-network / entity-gym
  Standard interface for entity based reinforcement learning environments.
  ☆35Updated 6 months ago
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆52Updated 2 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆39Updated 2 years ago
• RobertTLange / gymnax-blines
  Baselines for gymnax 🤖
  ☆57Updated last year
• DramaCow / jaxued
  ☆56Updated 3 weeks ago
• instadeepai / sebulba
  🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX
  ☆46Updated 10 months ago
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆52Updated last year
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆17Updated last month
• danijar / ninjax
  General Modules for JAX
  ☆57Updated last month
• keraJLi / synthetic-gymnax
  ☆17Updated 3 months ago
• roger-creus / Wave-Defense-Learning-Environment
  A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.
  ☆14Updated last year
• instadeepai / matrax
  A collection of matrix games in JAX
  ☆9Updated 8 months ago
• MichalBortkiewicz / JaxGCRL
  ☆25Updated last week
• EmptyJackson / groove
  Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…
  ☆22Updated 2 months ago
• AlexGoldie / rl-learned-optimization
  Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"
  ☆10Updated 2 months ago
• cswinter / DeepCodeCraft
  Deep reinforcement learning implementation that trains AIs for the CodeCraft real-time strategy game.
  ☆19Updated last year
• 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
• dibyaghosh / jaxrl_m
  Skeleton for scalable and flexible Jax RL implementations
  ☆58Updated last year
• vwxyzjn / gym-pysc2
  Gym wrapper for pysc2
  ☆10Updated 2 years ago
• MyNameIsArko / RL-Flax
  Various reinforcement learning algorithms written in Jax + Flax
  ☆21Updated last year
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆79Updated 2 years ago
• vwxyzjn / a2c_is_a_special_case_of_ppo
  A2C is a special case of PPO!
  ☆19Updated 2 years ago
• kenjyoung / mctx_learning_demo
  ☆46Updated last year
• eilab-gt / NovGrid
  Novelty MiniGrid--NovGrid--is an extension of MiniGrid environment that allows for the world properties and dynamics to change according …
  ☆35Updated 3 months ago
• icaros-usc / dqd-rl
  Official implementation of "Approximating Gradients for Differentiable Quality Diversity in Reinforcement Learning"
  ☆18Updated last year
• Howuhh / sac-n-jax
  Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch
  ☆46Updated last year
• MichaelTMatthews / Craftax_Baselines
  ☆14Updated last month
• hr0nix / omega
  A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…
  ☆37Updated 2 years ago
• ucl-dark / pax
  Scalable Opponent Shaping Experiments in JAX
  ☆19Updated 5 months ago
• Reytuag / transformerXL_PPO_JAX
  ☆59Updated last month