alexub / jax-meta-learningView external linksLinks
Simple, extensible implementations of some meta-learning algorithms in Jax
☆11Oct 6, 2020Updated 5 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:
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- A Jax based meta learning library☆16Sep 15, 2022Updated 3 years ago
- A collection of meta-learning algorithms in Jax☆24Sep 3, 2022Updated 3 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Mar 4, 2022Updated 3 years ago
- JAX implementations of core Deep RL algorithms☆83May 2, 2022Updated 3 years ago
- JAX/Flax implimentation of 'Attention Is All You Need' by Vaswani et al. (https://arxiv.org/abs/1706.03762)☆15Aug 16, 2021Updated 4 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 4 years ago
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 years ago
- ☆21Mar 14, 2021Updated 4 years ago
- Normalizing Flows using JAX☆86Nov 30, 2023Updated 2 years ago
- ☆23Oct 21, 2024Updated last year
- Implementation for Non-stationary Spectral Kernels (NIPS 2017)☆20Jan 4, 2020Updated 6 years ago
- Unofficial implementation of Siren with Jax for image representation.☆19Dec 10, 2021Updated 4 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 small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 2 years ago
- NeurIPS 2021 - Few-shot learning competition☆26Sep 27, 2021Updated 4 years ago
- How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…☆25Aug 16, 2023Updated 2 years ago
- Rethinking machine learning pipelines☆34Sep 29, 2025Updated 4 months ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆191Sep 28, 2022Updated 3 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) applied on Reinforcement Learning problems in TensorFlow 2.☆27May 11, 2021Updated 4 years ago
- Code to accompany paper 'Bayesian Deep Ensembles via the Neural Tangent Kernel'☆26Mar 12, 2021Updated 4 years ago
- A framework for composing Neural Processes in Julia☆76May 3, 2021Updated 4 years ago
- ☆12Jan 10, 2026Updated last month
- MetaLearners for CATE estimation☆48Feb 3, 2026Updated last week
- Energy-Based Hindsight Experience Prioritization (CoRL 2018) Oral presentation (7%)☆35Nov 28, 2018Updated 7 years ago
- Skeleton for scalable and flexible Jax RL implementations☆95Jul 1, 2023Updated 2 years ago
- Repository for the HPC & Data Science Summer Institute 2024☆12Mar 7, 2025Updated 11 months ago
- Water wave models in one dimension☆10Jan 4, 2026Updated last month
- Google Collab Notebooks for the UNIL Spring 2022 course on ML for Earth and Environmental Sciences☆14Aug 18, 2022Updated 3 years ago
- Learning Generalized Physical Representation from a Few Examples☆16Updated this week
- Source code and datasets for Circuit Design Completion using GNNs paper☆10Jan 26, 2023Updated 3 years ago
- ☆14Mar 3, 2022Updated 3 years ago
- Tutorial: Data Validation for Machine Learning Techniques☆11May 5, 2023Updated 2 years ago
- ☆40Jul 20, 2023Updated 2 years ago
- ☆164Dec 13, 2023Updated 2 years ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆50Jan 19, 2025Updated last year
- ☆40Jan 19, 2022Updated 4 years ago
- Goal-conditioned reinforcement learning like 🔥☆13Feb 3, 2024Updated 2 years ago