kenjyoung / mctx_learning_demoView external linksLinks
☆53Apr 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.☆64Nov 14, 2024Updated last year
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 2 years ago
- Monte Carlo tree search in JAX☆2,589Sep 2, 2025Updated 5 months ago
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆18Feb 27, 2023Updated 2 years ago
- Applying DeepMind's MuZero algorithm to the cart pole environment in gym☆21May 6, 2023Updated 2 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆23Oct 28, 2024Updated last year
- ☆22Nov 8, 2021Updated 4 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- ☆24Apr 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
- ☆13Jul 12, 2024Updated last year
- Simple PyTorch profiler that combines DeepSpeed Flops Profiler and TorchInfo☆11Feb 12, 2023Updated 3 years ago
- ☆12Mar 12, 2024Updated last year
- Made for a reading group at the Center for Safe AGI.☆12Oct 27, 2022Updated 3 years ago
- On the Feasibility of Cross-Task Transfer with Model-Based Reinforcement Learning☆16Apr 30, 2023Updated 2 years ago
- ☆16Jul 16, 2024Updated last year
- 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
- Modified version of the LeagueSandbox project which relies on a Redis server to accept actions and send observations. Intended for reinfo…☆11Feb 23, 2025Updated 11 months ago
- ☆14Oct 11, 2022Updated 3 years ago
- ☆12Apr 22, 2022Updated 3 years ago
- Extending rllab to event-driven multiagent environments☆13Oct 1, 2018Updated 7 years ago
- ☆37Apr 27, 2023Updated 2 years ago
- I publish my weekly research here☆20Jun 26, 2025Updated 7 months ago
- ☆19Jan 16, 2025Updated last year
- ☆17Mar 21, 2021Updated 4 years ago
- A collection of matrix games in JAX☆13Nov 28, 2024Updated last year
- Soccer toy example simulator used in Reinforcement Learning☆12Mar 11, 2018Updated 7 years ago
- Code for reproducing our paper "Low Rank Adapting Models for Sparse Autoencoder Features"☆17Mar 31, 2025Updated 10 months ago
- A minimal PyTorch re-implementation of GPT (Generative Pretrained Transformer) language model training☆18Sep 15, 2023Updated 2 years ago
- Model-based reinforcement learning using CEM, MPC and PETS☆16Nov 20, 2019Updated 6 years ago
- A reinforcement learning environment for the IGLU 2022 at NeurIPS☆35May 27, 2023Updated 2 years ago