kenjyoung / mctx_learning_demoLinks
β51Updated 2 years ago
Alternatives and similar repositories for mctx_learning_demo
Users that are interested in mctx_learning_demo are comparing it to the libraries listed below
Sorting:
- Baselines for gymnax π€β66Updated 2 years ago
- A project that provides help for using DeepMind's mctx on gym-style environments.β60Updated 6 months ago
- Classic MCTS example with mctxβ18Updated 2 years ago
- General Modules for JAXβ66Updated last month
- β79Updated 2 months ago
- An implementation of MuZero in JAX.β56Updated 2 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learningβ73Updated 9 months ago
- JAX implementations of core Deep RL algorithmsβ79Updated 3 years ago
- AlphaZero in JAXβ77Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weightsβ¦β56Updated 2 years ago
- A collection of RL algorithms written in JAX.β97Updated 2 years ago
- Accelerated replay buffers in JAXβ41Updated 2 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselinβ¦β53Updated 3 weeks ago
- Code for magnetic mirror descent.β16Updated last year
- Vectorization techniques for fast population-based training.β56Updated 2 years ago
- fast + parallel AlphaZero in JAXβ96Updated 5 months ago
- Reinforcement learning training framework for entity-gym environments.β17Updated last year
- Code for Discovered Policy Optimisation (NeurIPS 2022)β10Updated last year
- Monte Carlo tree search in JAX, with functionality to continue search from a previous subtreeβ20Updated last month
- Standard interface for entity based reinforcement learning environments.β37Updated last year
- Novelty MiniGrid--NovGrid--is an extension of MiniGrid environment that allows for the world properties and dynamics to change according β¦β35Updated last year
- Reinforcement learning library in JAX.β100Updated last year
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papersβ52Updated 2 years ago
- β80Updated 7 months ago
- Various reinforcement learning algorithms written in Jax + Flaxβ24Updated last year
- β56Updated 2 years ago
- An Open-Ended Agentic Simulatorβ51Updated 9 months ago
- Accelerated minigrid environments with JAXβ138Updated 2 weeks ago
- Simple single-file baselines for Q-Learning in pure-GPU settingβ167Updated 2 months ago
- Deep Reinforcement Learning Framework done with PyTorchβ36Updated 2 months ago