mohmdelsayed / HesScale
Scalable Computation of Hessian Diagonals
☆13Updated 11 months ago
Alternatives and similar repositories for HesScale
Users that are interested in HesScale are comparing it to the libraries listed below
Sorting:
- Automatically generate simple meta-learning tasks from a very large space☆15Updated last year
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆28Updated 2 years ago
- Generalised UDRL☆37Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆19Updated 4 years ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- TaskMet Task-driven Metric Learning for Model Learning☆19Updated last year
- Repo to reproduce the First-Explore paper results☆37Updated 4 months ago
- Neuroevolution Benchmark in JAX 🦕☆38Updated last year
- Drop-in environment replacements that make your RL algorithm train faster.☆20Updated 10 months ago
- ☆31Updated last year
- PyTorch Package For Quasimetric Learning☆42Updated 6 months ago
- A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. a…☆21Updated 4 years ago
- Variational Reinforcement Learning☆16Updated 9 months ago
- AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)☆18Updated 7 months ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated last year
- Scalable Opponent Shaping Experiments in JAX☆24Updated last year
- Simple, extensible implementations of some meta-learning algorithms in Jax☆10Updated 4 years ago
- ☆13Updated 10 months ago
- ☆9Updated 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
- Online variational GPs☆33Updated last year
- flexible meta-learning in jax☆13Updated last year
- ☆31Updated 4 years ago
- Official repository for the paper "Automating Continual Learning"☆14Updated last year
- Official codebase for Improving Computational Efficiency in Visual Reinforcement Learning via Stored Embeddings.☆21Updated 4 years ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Updated 3 years ago
- Code that can be used to reproduce the experiments in our paper "Estimating Risk and Uncertainty in Deep Reinforcement Learning"☆30Updated 2 years ago
- Official pytorch implementation for our ICLR 2023 paper "Latent State Marginalization as a Low-cost Approach for Improving Exploration".☆24Updated 2 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆10Updated last year
- Supplementary Data for Evolving Reinforcement Learning Algorithms☆46Updated 4 years ago