levilelis / h-levin
Levin tree search guided by both a policy and a heuristic function
☆18Updated last year
Alternatives and similar repositories for h-levin:
Users that are interested in h-levin are comparing it to the libraries listed below
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆41Updated 2 years ago
- General Modules for JAX☆64Updated last month
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆31Updated 3 years ago
- An implementation of MuZero in JAX.☆56Updated 2 years ago
- Neuroevolution Benchmark in JAX 🦕☆38Updated last year
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- Monte Carlo tree search in JAX, with functionality to continue search from a previous subtree☆19Updated last week
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆28Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆54Updated 2 years ago
- ☆77Updated last month
- ☆53Updated 6 months ago
- PyTorch Package For Quasimetric Learning☆42Updated 6 months ago
- A project that provides help for using DeepMind's mctx on gym-style environments.☆59Updated 5 months ago
- flexible meta-learning in jax☆13Updated last year
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Function☆13Updated 2 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆50Updated last week
- Learning diverse options through the Laplacian representation.☆23Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆100Updated last year
- ☆20Updated 2 years ago
- Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…☆26Updated 10 months ago
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆22Updated 2 years ago
- Scalable Opponent Shaping Experiments in JAX☆24Updated last year
- On the model-based stochastic value gradient for continuous reinforcement learning☆55Updated last year
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Updated 3 years ago
- ☆14Updated 5 years ago
- Vectorization techniques for fast population-based training.☆56Updated 2 years ago
- ☆20Updated 5 years ago
- Reinforcement learning in pure JAX.☆12Updated 2 months ago
- Scaling scaling laws with board games.☆48Updated last year
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆10Updated last year