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 4 years ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 3 years ago
- Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API☆13Sep 25, 2021Updated 4 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
- Normalizing Flows using JAX☆86Nov 30, 2023Updated 2 years ago
- ☆20Mar 14, 2021Updated 4 years ago
- ☆24Oct 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 3 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
- Maximum Entropy-Regularized Multi-Goal Reinforcement Learning (ICML 2019)☆24May 30, 2019Updated 6 years ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 5 years ago
- Rethinking machine learning pipelines☆34Sep 29, 2025Updated 5 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
- 💧 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
- ☆12Jan 10, 2026Updated last month
- A framework for composing Neural Processes in Julia☆76May 3, 2021Updated 4 years ago
- MetaLearners for CATE estimation☆49Updated this 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☆96Jul 1, 2023Updated 2 years ago
- Repository for the HPC & Data Science Summer Institute 2024☆12Mar 7, 2025Updated last year
- Code for utilising VAE as means of doing exact MCMC inference in complex high-dimensional space☆14Jun 20, 2023Updated 2 years ago
- Water wave models in one dimension☆10Feb 24, 2026Updated last week
- ☆14Mar 3, 2022Updated 4 years ago
- This is the official codebase for paper: Scaling Verification Can Be More Effective than Scaling Policy Learning for Vision-Language-Acti…☆39Feb 24, 2026Updated last week
- R&D Python package for processing ICON data with xarray☆10Jul 26, 2023Updated 2 years ago
- Source code and datasets for Circuit Design Completion using GNNs paper☆10Jan 26, 2023Updated 3 years ago
- Google Collab Notebooks for the UNIL Spring 2022 course on ML for Earth and Environmental Sciences☆14Aug 18, 2022Updated 3 years ago