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 3 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years 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.
- A collection of matrix games in JAX☆13Nov 28, 2024Updated last year
- Datasets for compositional learning☆11Nov 28, 2018Updated 7 years ago
- ☆23Dec 7, 2025Updated 3 months ago
- Accelerated minigrid environments with JAX☆163Oct 20, 2025Updated 5 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
- ☆18Dec 10, 2023Updated 2 years ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated last year
- Minimal A2C/A3C example of an LSTM-based meta-learner.☆13Feb 2, 2021Updated 5 years ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆15Jan 3, 2023Updated 3 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- ☆13Aug 4, 2022Updated 3 years ago
- 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
- Reinforcement Learning inside a 3D soccer simulation☆37Sep 15, 2024Updated last year
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆33Jan 9, 2022Updated 4 years ago
- A collection of meta-learning algorithms in Jax☆24Sep 3, 2022Updated 3 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆112Dec 5, 2023Updated 2 years ago
- The code for experiments conducted to verify the correctness of mirror learning.☆11Jun 3, 2022Updated 3 years ago
- PushWorld: A benchmark for manipulation planning with tools and movable obstacles☆90Mar 12, 2026Updated 2 weeks ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Baselines for gymnax 🤖☆75Apr 3, 2023Updated 2 years ago
- a happy approach to online behavioral research☆19Mar 8, 2026Updated 2 weeks ago
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025☆82Feb 13, 2025Updated last year
- A simple hypernetwork implementation in jax using haiku.☆24Aug 20, 2022Updated 3 years ago
- Randomized Linear Algebra in Python☆13Mar 21, 2017Updated 9 years ago
- Efficient baselines for autocurricula in JAX.☆209Aug 24, 2024Updated last year
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- JAX implementation of Instant-NGP (NeRF part)☆37May 25, 2024Updated last year
- Reimplementation of ToMNet with some extensions for RL as well☆14Apr 28, 2018Updated 7 years ago
- NordVPN Threat Protection Pro™ • AdTake your cybersecurity to the next level. Block phishing, malware, trackers, and ads. Lightweight app that works with all browsers.
- Implements the Messenger environment and EMMA model.☆25Jun 14, 2023Updated 2 years ago
- PYTORCH Implementation of the supervised learning experiments in Vector-based navigation using grid-like representations in artificial ag…☆19May 22, 2020Updated 5 years ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆274Sep 22, 2025Updated 6 months ago
- Official code for the paper "Attention as a Hypernetwork"☆55Feb 24, 2026Updated last month
- Neural model of hierarchical reinforcement learning☆16Sep 14, 2017Updated 8 years ago
- Jax implementation of VIT-VQGAN☆10Jan 25, 2024Updated 2 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago