A collection of meta-learning algorithms in Jax
☆24Sep 3, 2022Updated 3 years ago
Alternatives and similar repositories for jax-meta-learning
Users that are interested in jax-meta-learning are comparing it to the libraries listed below
Sorting:
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- A Jax based meta learning library☆16Sep 15, 2022Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- flexible meta-learning in jax☆16Oct 19, 2023Updated 2 years ago
- Pytrees + dataclasses ❤️☆75Dec 19, 2025Updated 2 months ago
- ☆17Aug 3, 2022Updated 3 years ago
- Train very large language models in Jax.☆210Oct 21, 2023Updated 2 years ago
- The original code for the paper "Learning to Learn via Self-Critique".☆48Aug 10, 2019Updated 6 years ago
- Official implementation of the δ-model presented in the ICML 2024 paper "A Distributional Analogue to the Successor Representation".☆25Nov 8, 2024Updated last year
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Mar 4, 2022Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 3 years ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆62Sep 3, 2025Updated 6 months ago
- A from-scratch neural network and transformers library, with speeds rivaling PyTorch☆10Mar 16, 2025Updated 11 months ago
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 3 months ago
- ☆12Jan 10, 2026Updated 2 months ago
- 💧 Inland water systems are essential to our environment because they are vital ecosystems that are bio-diverse. Thus, finding innovative…☆11Apr 26, 2020Updated 5 years ago
- [ICLR 2026] ParallelBench: Understanding the Tradeoffs of Parallel Decoding in Diffusion LLMs☆42Updated this week
- This is the public repo for the course HMMA238 'Software Development'☆10Apr 20, 2021Updated 4 years ago
- ☆30Jan 19, 2023Updated 3 years ago
- Software dev. for data science (Python)☆15May 1, 2025Updated 10 months ago
- A Julia package for adaptive proximal gradient and primal-dual algorithms☆11Jan 18, 2024Updated 2 years 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 3 years ago
- Scaling safe exploration to vision control☆14Feb 19, 2025Updated last year
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆273Sep 22, 2025Updated 5 months ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆396Mar 1, 2026Updated last week
- [ICLR'25] "Understanding Bottlenecks of State Space Models through the Lens of Recency and Over-smoothing" by Peihao Wang, Ruisi Cai, Yue…☆17Mar 21, 2025Updated 11 months ago
- Attempt at cog wrapper for segmind/SSD-1B☆10Dec 11, 2023Updated 2 years ago
- Official implementation of Tabular Transfer Learning via Prompting LLMs (COLM 2024).☆13Aug 6, 2024Updated last year
- ☆11Apr 14, 2022Updated 3 years ago
- MMD-FUSE package implementing the MMD-FUSE test proposed in MMD-FUSE: Learning and Combining Kernels for Two-Sample Testing Without Data …☆11May 31, 2024Updated last year
- ☆12Oct 8, 2020Updated 5 years ago
- ☆18Jul 12, 2024Updated last year
- Learning to draw samples: with application to amortized maximum likelihood estimator for generative adversarial learning☆10Dec 28, 2021Updated 4 years ago
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Function☆13Nov 22, 2022Updated 3 years ago
- Test equality between a black-box LLM API and a reference distribution☆12Oct 29, 2024Updated last year
- Goal-conditioned reinforcement learning like 🔥☆13Feb 3, 2024Updated 2 years ago
- State-of-The-Art Rating-based RECOmmendation system: pytorch lightning implementation☆13Oct 10, 2023Updated 2 years ago
- ☆11Jul 13, 2018Updated 7 years ago