☆54Apr 11, 2023Updated 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:
- AlphaZero in JAX☆81Apr 3, 2024Updated last year
- Classic MCTS example with mctx☆24May 25, 2023Updated 2 years ago
- A project that provides help for using DeepMind's mctx on gym-style environments.☆65Nov 14, 2024Updated last year
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 2 years ago
- Adding Dreamer-v3's implementation tricks to CleanRL's PPO☆14May 19, 2023Updated 2 years ago
- Monte Carlo tree search in JAX☆2,596Sep 2, 2025Updated 6 months ago
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆18Feb 27, 2023Updated 3 years ago
- Applying DeepMind's MuZero algorithm to the cart pole environment in gym☆21May 6, 2023Updated 2 years ago
- ☆22Nov 8, 2021Updated 4 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆23Oct 28, 2024Updated last year
- Monte Carlo tree search in JAX, with functionality to continue search from a previous subtree☆26May 2, 2025Updated 10 months ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- ☆25Apr 16, 2024Updated last year
- Swarm learning algorithm☆11Jun 2, 2021Updated 4 years ago
- JAX/Haiku implementation of "Auction Learning as a Two-Player Game"☆11Jul 6, 2024Updated last year
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆43Sep 19, 2022Updated 3 years ago
- Baselines for gymnax 🤖☆74Apr 3, 2023Updated 2 years ago
- fast + parallel AlphaZero in JAX☆109Dec 22, 2024Updated last year
- Simple PyTorch profiler that combines DeepSpeed Flops Profiler and TorchInfo☆11Feb 12, 2023Updated 3 years ago
- ☆12Mar 12, 2024Updated last year
- ☆13Jul 12, 2024Updated last year
- Made for a reading group at the Center for Safe AGI.☆12Feb 23, 2026Updated last week
- On the Feasibility of Cross-Task Transfer with Model-Based Reinforcement Learning☆16Apr 30, 2023Updated 2 years ago
- ☆16Jul 16, 2024Updated last year
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- JAX-based implementation for multi-agent path planning (MAPP) in continuous spaces.☆54Dec 13, 2022Updated 3 years ago
- Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.☆138Aug 20, 2024Updated last year
- An implementation of MuZero in JAX.☆57Nov 8, 2022Updated 3 years ago
- Extending rllab to event-driven multiagent environments☆13Oct 1, 2018Updated 7 years ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆15Jan 3, 2023Updated 3 years ago
- ☆14Oct 11, 2022Updated 3 years ago
- ☆12Apr 22, 2022Updated 3 years ago
- Modified version of the LeagueSandbox project which relies on a Redis server to accept actions and send observations. Intended for reinfo…☆11Feb 23, 2025Updated last year
- ☆37Apr 27, 2023Updated 2 years ago
- Accompanying code for "Learning and Planning in Average-Reward Markov Decision Processes"☆15Feb 10, 2021Updated 5 years ago
- ☆17Mar 21, 2021Updated 4 years ago
- I publish my weekly research here☆20Jun 26, 2025Updated 8 months ago
- ☆19Jan 16, 2025Updated last year
- A collection of matrix games in JAX☆13Nov 28, 2024Updated last year