flexible meta-learning in jax
☆16Oct 19, 2023Updated 2 years ago
Alternatives and similar repositories for metax
Users that are interested in metax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Code accompanying the paper "A contrastive rule for meta-learning"☆13Oct 31, 2024Updated last year
- JAX implementation of the Mistral 7b v0.1 model☆13Mar 27, 2024Updated 2 years ago
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 7 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 3 years ago
- A collection of matrix games in JAX☆14Apr 13, 2026Updated 3 months ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- Datasets for compositional learning☆11Nov 28, 2018Updated 7 years ago
- Accelerated minigrid environments with JAX☆175Oct 20, 2025Updated 9 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…☆23Dec 22, 2020Updated 5 years ago
- Official code for the paper "Attention as a Hypernetwork"☆58Feb 24, 2026Updated 5 months ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆24Feb 15, 2025Updated last year
- ☆20Dec 10, 2023Updated 2 years ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆14Jan 3, 2023Updated 3 years ago
- Minimal A2C/A3C example of an LSTM-based meta-learner.☆13Feb 2, 2021Updated 5 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimization☆25Dec 1, 2025Updated 7 months ago
- ☆13Aug 4, 2022Updated 3 years ago
- Reinforcement Learning inside a 3D soccer simulation☆37Sep 15, 2024Updated last year
- Code for enumeration from a CFG☆20Sep 25, 2023Updated 2 years ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆61Oct 23, 2023Updated 2 years ago
- Code accompanying the paper "Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysis"☆15Mar 21, 2024Updated 2 years ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆33Jan 9, 2022Updated 4 years ago
- a happy approach to online behavioral research☆19Mar 25, 2026Updated 4 months ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆116Dec 5, 2023Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A collection of meta-learning algorithms in Jax☆24Sep 3, 2022Updated 3 years ago
- The code for experiments conducted to verify the correctness of mirror learning.☆11Jun 3, 2022Updated 4 years ago
- PushWorld: A benchmark for manipulation planning with tools and movable obstacles☆95May 5, 2026Updated 2 months ago
- A simple hypernetwork implementation in jax using haiku.☆24Aug 20, 2022Updated 3 years ago
- Baselines for gymnax 🤖☆78Apr 3, 2023Updated 3 years ago
- Convert jax functions to sympy for symbolic analysis. This is useful for sparsity detection.☆21Jul 15, 2026Updated last week
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025☆84Feb 13, 2025Updated last year
- ☆25Apr 27, 2026Updated 3 months ago
- Randomized Linear Algebra in Python☆13Mar 21, 2017Updated 9 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Simple Successor Features☆20Jul 15, 2025Updated last year
- Efficient baselines for autocurricula in JAX.☆214Aug 24, 2024Updated last year
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- Reimplementation of ToMNet with some extensions for RL as well☆14Apr 28, 2018Updated 8 years ago
- Implements the Messenger environment and EMMA model.☆25Jun 14, 2023Updated 3 years ago
- PYTORCH Implementation of the supervised learning experiments in Vector-based navigation using grid-like representations in artificial ag…☆19May 22, 2020Updated 6 years ago
- JAX implementation of Instant-NGP (NeRF part)☆37May 25, 2024Updated 2 years ago