smonsays / metaxView external linksLinks
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
Sorting:
- JAX implementation of the Mistral 7b v0.1 model☆13Mar 27, 2024Updated last year
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 2 months ago
- Code accompanying the paper "A contrastive rule for meta-learning"☆13Oct 31, 2024Updated last year
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 2 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
- A collection of matrix games in JAX☆13Nov 28, 2024Updated last year
- Accelerated minigrid environments with JAX☆156Oct 20, 2025Updated 3 months ago
- A collection of meta-learning algorithms in Jax☆24Sep 3, 2022Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated 11 months ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆61Oct 23, 2023Updated 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…☆23Dec 22, 2020Updated 5 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆260Oct 31, 2025Updated 3 months ago
- Implements the Messenger environment and EMMA model.☆25Jun 14, 2023Updated 2 years ago
- a modular reinforcement learning library with JAX agents☆27Mar 3, 2025Updated 11 months 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
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆270Sep 22, 2025Updated 4 months ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆36May 19, 2023Updated 2 years ago
- Reinforcement Learning inside a 3D soccer simulation☆36Sep 15, 2024Updated last year
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- Efficient baselines for autocurricula in JAX.☆206Aug 24, 2024Updated last year
- DreamerV3 implementation of Curious Replay, a method for prioritizing experience replay that is tailored to model-based reinforcement lea…☆37Jul 5, 2023Updated 2 years ago
- A Gymnasium environment for simulating multi-robot planning.☆30Sep 23, 2023Updated 2 years ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 2 years ago
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025☆81Feb 13, 2025Updated last year
- Open source code for EigenGame.☆34May 15, 2023Updated 2 years ago
- Baselines for gymnax 🤖☆74Apr 3, 2023Updated 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
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 2 years ago
- The official starter-kit for NeurIPS 2025 mind games competition☆21Jul 27, 2025Updated 6 months ago
- PushWorld: A benchmark for manipulation planning with tools and movable obstacles☆90Jan 13, 2026Updated last month
- ☆47Nov 26, 2025Updated 2 months ago
- ☆35Nov 22, 2024Updated last year
- Partially Observable Multi-Agent RL with Transformers☆17Updated this week
- Navier-Stokes implementation for Processing based on Jos Stam work☆13Feb 1, 2019Updated 7 years ago
- Embedding language models in probability space via log-likelihood vectors☆16Oct 25, 2025Updated 3 months ago
- Python tools for running xfoil many times☆10Sep 10, 2016Updated 9 years ago
- Lightweight Neural Architecture Search for Temporal Convolutional Networks at the Edge☆10Mar 6, 2023Updated 2 years ago
- ☆15Jul 27, 2023Updated 2 years ago