flexible meta-learning in jax
☆17Oct 19, 2023Updated 3 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 2 years ago
- JAX implementation of the Mistral 7b v0.1 model☆13Mar 27, 2024Updated 2 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 11 months ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 6 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- A JAX-Native Interface for Reinforcement Learning Environments☆10Oct 6, 2025Updated last year
- Datasets for compositional learning☆11Nov 28, 2018Updated 7 years ago
- Offshore asset lifecycle: ASCII data → strength, fatigue, FFS, hydrodynamics☆18Updated this week
- A collection of matrix games in JAX☆16Apr 13, 2026Updated 6 months ago
- Official code for the paper "Attention as a Hypernetwork"☆60Feb 24, 2026Updated 8 months ago
- Official code repository for the publication "Credit Assignment in Neural Networks through Deep Feedback Control".☆12Oct 31, 2024Updated 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
- A fast implementation of MAML (Finn et al. 2017) in Tensorflow 2☆12Jul 30, 2021Updated 5 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
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Minimal A2C/A3C example of an LSTM-based meta-learner.☆13Feb 2, 2021Updated 5 years ago
- ☆13Aug 4, 2022Updated 4 years ago
- Accelerated minigrid environments with JAX☆184Sep 10, 2026Updated last month
- Code accompanying the paper "Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysis"☆14Mar 21, 2024Updated 2 years ago
- Code for enumeration from a CFG☆20Sep 25, 2023Updated 3 years ago
- Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimization☆30Dec 1, 2025Updated 11 months ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆25Feb 15, 2025Updated last year
- ☆20Dec 10, 2023Updated 2 years ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆34Jan 9, 2022Updated 4 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Adaptive, optimal, and learning-based control for quadrotor swarms using gym-pybullet-drones.☆17Sep 16, 2025Updated last year
- A simple hypernetwork implementation in jax using haiku.☆24Aug 20, 2022Updated 4 years ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆62Aug 22, 2026Updated 2 months ago
- 🫧 Differentiable microbubble dynamics in JAX.☆17Updated this week
- The code for experiments conducted to verify the correctness of mirror learning.☆11Jun 3, 2022Updated 4 years ago
- Reinforcement Learning inside a 3D soccer simulation☆35Sep 15, 2024Updated 2 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆118Dec 5, 2023Updated 2 years ago
- PushWorld: A benchmark for manipulation planning with tools and movable obstacles☆96May 5, 2026Updated 5 months ago
- A prettier rocm-smi output with color-coded GPU stats☆11Feb 19, 2026Updated 8 months 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 4 years ago
- Convert jax functions to sympy for symbolic analysis. This is useful for sparsity detection.☆22Jul 15, 2026Updated 3 months ago
- Randomized Linear Algebra in Python☆13Mar 21, 2017Updated 9 years ago
- Baselines for gymnax 🤖☆82Apr 3, 2023Updated 3 years ago
- Collection of resources from prior speaking engagements and guidance on booking Chinasa for speaking engagements.☆12Oct 12, 2026Updated 3 weeks ago
- Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…☆36Sep 13, 2026Updated last month
- SmolWAM: a 0.88B latent world-action model for robot manipulation on edge devices. NeurIPS 2026 Workshop on Robotics World Modeling.☆14Oct 9, 2026Updated 3 weeks ago