mohmdelsayed / HesScaleLinks
Scalable Computation of Hessian Diagonals
☆13Updated last year
Alternatives and similar repositories for HesScale
Users that are interested in HesScale are comparing it to the libraries listed below
Sorting:
- ☆13Updated last year
- 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
- 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…☆22Updated 4 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
- Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…☆30Updated last year
- Drop-in environment replacements that make your RL algorithm train faster.☆21Updated last year
- Neuroevolution Benchmark in JAX 🦕☆39Updated last year
- A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks☆33Updated 8 months ago
- ☆31Updated 2 years ago
- Repo to reproduce the First-Explore paper results☆37Updated 6 months ago
- Implementation of Proximal Policy Optimization in Jax+Flax☆20Updated 2 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆11Updated 2 years ago
- flexible meta-learning in jax☆14Updated last year
- Scaling scaling laws with board games.☆49Updated 2 years ago
- Neuroevolution is a Competitive Alternative to Reinforcement Learning for Skill Discovery☆16Updated 2 years ago
- TaskMet Task-driven Metric Learning for Model Learning☆19Updated last year
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆12Updated this week
- Clockwork VAEs in JAX/Flax☆32Updated 4 years ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated 2 years ago
- Supplementary Data for Evolving Reinforcement Learning Algorithms☆46Updated 4 years ago
- ☆54Updated 8 months ago
- ☆32Updated last year
- ☆23Updated 3 years ago
- GPT implementation in Flax☆18Updated 3 years ago
- Scalable Opponent Shaping Experiments in JAX☆24Updated last year
- Code for minimum-entropy coupling.☆32Updated last year
- Learn online intrinsic rewards from LLM feedback☆41Updated 7 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆84Updated last year