Alternatives and similar repositories for omega

Users that are interested in omega are comparing it to the libraries listed below

• Hwhitetooth / jax_muzero

  View on GitHub
  An implementation of MuZero in JAX.
  ☆57Nov 8, 2022Updated 3 years ago
• luchris429 / discovered-policy-optimisation

  View on GitHub
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆12Jun 15, 2023Updated 2 years ago
• tuero / muzero-cpp

  View on GitHub
  A C++ pytorch implementation of MuZero
  ☆40May 1, 2024Updated last year
• DHDev0 / Muzero-unplugged

  View on GitHub
  Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observ…
  ☆35Jun 25, 2025Updated 7 months ago
• DHDev0 / Stochastic-muzero

  View on GitHub
  Pytorch Implementation of Stochastic MuZero for gym environment. This algorithm is capable of supporting a wide range of action and obser…
  ☆75Dec 31, 2025Updated last month
• Carbon225 / mctx-classic

  View on GitHub
  Classic MCTS example with mctx
  ☆24May 25, 2023Updated 2 years ago
• fidel-schaposnik / muzero

  View on GitHub
  Tensorflow implementation of MuZero algorithm
  ☆11Aug 23, 2022Updated 3 years ago
• kaesve / muzero

  View on GitHub
  A clean implementation of MuZero and AlphaZero following the AlphaZero General framework. Train and Pit both algorithms against each othe…
  ☆168Mar 28, 2021Updated 4 years ago
• opooladz / Preconditioned-Stochastic-Gradient-Descent

  View on GitHub
  A repo based on XiLin Li's PSGD repo that extends some of the experiments.
  ☆14Oct 7, 2024Updated last year
• cswinter / hyperstate

  View on GitHub
  Opinionated library for managing hyperparameters and mutable state of machine learning training systems.
  ☆19Aug 4, 2023Updated 2 years ago
• RPegoud / jym

  View on GitHub
  JAX implementation of RL algorithms and vectorized environments
  ☆51Dec 26, 2023Updated 2 years ago
• bwfbowen / muax

  View on GitHub
  A project that provides help for using DeepMind's mctx on gym-style environments.
  ☆64Nov 14, 2024Updated last year
• RyanNavillus / PPO-v3

  View on GitHub
  Adding Dreamer-v3's implementation tricks to CleanRL's PPO
  ☆14May 19, 2023Updated 2 years ago
• mjanschek / pytorch_seed_rl

  View on GitHub
  A PyTorch implementation of SEED, originally created by Google Research for TensorFlow 2.
  ☆15Dec 8, 2020Updated 5 years ago
• symoon11 / dreamerv3-flax

  View on GitHub
  Flax Implementation of DreamerV3 on Crafter
  ☆18Nov 29, 2025Updated 2 months ago
• YeWR / EfficientZero

  View on GitHub
  Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.
  ☆923Dec 20, 2023Updated 2 years ago
• ben-eysenbach / mnm

  View on GitHub
  Code to accompany the paper "Mismatched No More: Joint Model-Policy Optimization for Model-Based RL"
  ☆20Oct 6, 2021Updated 4 years ago
• sdpkjc / abcdrl

  View on GitHub
  Modular Single-file Reinfocement Learning Algorithms Library
  ☆38May 16, 2023Updated 2 years ago
• JimOhman / model-based-rl

  View on GitHub
  Implementation of MuZero with PyTorch, based on the pseudocode from DeepMind (https://arxiv.org/src/1911.08265v2/anc/pseudocode.py).
  ☆33Aug 14, 2022Updated 3 years ago
• amidos2006 / pcg_benchmark

  View on GitHub
  A benchmark to test and compare your pcg algorithm against each other
  ☆22Sep 22, 2025Updated 4 months ago
• subho406 / Recurrent-PPO-Jax

  View on GitHub
  Implementation of Proximal Policy Optimization in Jax+Flax
  ☆21May 18, 2023Updated 2 years ago
• google-deepmind / nao_top10

  View on GitHub
  ☆19Mar 1, 2023Updated 2 years ago
• NTT123 / a0-jax

  View on GitHub
  AlphaZero in JAX
  ☆81Apr 3, 2024Updated last year
• brownirl / lambda_discrepancy

  View on GitHub
  Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy
  ☆23Oct 28, 2024Updated last year
• lowrollr / mctx-az

  View on GitHub
  Monte Carlo tree search in JAX, with functionality to continue search from a previous subtree
  ☆26May 2, 2025Updated 9 months ago
• hamishs / JAX-RL

  View on GitHub
  JAX implementations of various deep reinforcement learning algorithms.
  ☆26Feb 2, 2025Updated last year
• gcucurull / maml_flax

  View on GitHub
  Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.
  ☆21Sep 18, 2020Updated 5 years ago
• sash-a / CleanRL.jl

  View on GitHub
  Simple single file implementations of Reinforcement Learning algorithms in Julia
  ☆23Feb 15, 2025Updated 11 months ago
• koulanurag / muzero-pytorch

  View on GitHub
  Pytorch Implementation of MuZero
  ☆352Jul 23, 2023Updated 2 years ago
• corl-team / katakomba

  View on GitHub
  Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)
  ☆43Aug 22, 2023Updated 2 years ago
• dschaurecker / bitepy

  View on GitHub
  A Battery Intraday Trading Engine, based on dynamic programming approximations, written in C++, wrapped for Python
  ☆32Feb 5, 2026Updated last week
• aliang8 / varibad_jax

  View on GitHub
  ☆10Jun 27, 2024Updated last year
• davisyoshida / jax-gptq

  View on GitHub
  JAX implementation of GPTQ quantization algorithm
  ☆10Jul 19, 2023Updated 2 years ago
• sculd / algorithmic_intraday_trading

  View on GitHub
  ☆11Oct 6, 2020Updated 5 years ago
• RobertTLange / gymnax-blines

  View on GitHub
  Baselines for gymnax 🤖
  ☆74Apr 3, 2023Updated 2 years ago
• kronion / gym-sts

  View on GitHub
  Gym env for Slay the Spire
  ☆16Dec 31, 2024Updated last year
• nmonette / NCC-UED

  View on GitHub
  Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025
  ☆17Nov 24, 2025Updated 2 months ago
• kelechi-c / dit_flow

  View on GitHub
  DiT (training + flow matching) in Jax
  ☆11Jan 5, 2025Updated last year
• uoe-agents / CMID

  View on GitHub
  ☆13Apr 25, 2024Updated last year