Simple, extensible implementations of some meta-learning algorithms in Jax
☆11Oct 6, 2020Updated 6 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 6 years ago
- A Jax based meta learning library☆15Sep 15, 2022Updated 4 years ago
- A collection of meta-learning algorithms in Jax☆24Sep 3, 2022Updated 4 years ago
- flexible meta-learning in jax☆17Oct 19, 2023Updated 3 years ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Mar 4, 2022Updated 4 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Adaptable Agent Populations via a Generative Model of Policies☆12Oct 14, 2021Updated 5 years ago
- JAX/Flax implimentation of 'Attention Is All You Need' by Vaswani et al. (https://arxiv.org/abs/1706.03762)☆15Aug 16, 2021Updated 5 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 5 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Implementation for Non-stationary Spectral Kernels (NIPS 2017)☆20Jan 4, 2020Updated 6 years ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 years ago
- Unofficial implementation of Siren with Jax for image representation.☆19Dec 10, 2021Updated 4 years ago
- Implementation of Model-Agnostic Meta-Learning (MAML) in Jax☆192Sep 28, 2022Updated 4 years ago
- Maximum Entropy-Regularized Multi-Goal Reinforcement Learning (ICML 2019)☆24May 30, 2019Updated 7 years 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 framework for composing Neural Processes in Julia☆74May 3, 2021Updated 5 years ago
- Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API☆13Sep 25, 2021Updated 5 years ago
- Convert jax functions to sympy for symbolic analysis. This is useful for sparsity detection.☆22Jul 15, 2026Updated 3 months ago
- 3D geoms for plotnine (grammar of graphics in Python)☆14Aug 5, 2022Updated 4 years ago
- Causal estimators for use with WhyNot☆11Mar 20, 2020Updated 6 years ago
- ☆21Mar 14, 2021Updated 5 years ago
- Latex package for drawing (causal) influence diagrams (CIDs) and labeling incentives, maintained by the Causal Incentives Working Group☆13Mar 1, 2021Updated 5 years ago
- Code for Deep Structured Mixtures of Gaussian Processes (DSMGPs)☆11Jan 27, 2022Updated 4 years ago
- R&D Python package for processing ICON data with xarray☆10Jul 26, 2023Updated 3 years 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.
- ☆11Apr 2, 2022Updated 4 years ago
- Multi-output Gaussian processes in JAX.☆10Dec 31, 2022Updated 3 years ago
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 years ago
- Prototype for sparse automatic differentiation in JAX + Julia☆24Jun 17, 2025Updated last year
- keras implementation of wavenet/parallel wavenet☆10Jun 16, 2019Updated 7 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Dec 21, 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 3 years ago
- Code for utilising VAE as means of doing exact MCMC inference in complex high-dimensional space☆14Jun 20, 2023Updated 3 years ago
- Normalizing Flows using JAX☆87Nov 30, 2023Updated 2 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- A collection of Pythonic subword tokenisers and text preprocessing tools.☆15Oct 29, 2026Updated last week
- Create and update Anki cards for Japanese☆12Jan 19, 2025Updated last year
- ☆17Oct 19, 2020Updated 6 years ago
- Ἀνατομή is a PyTorch library to analyze representation of neural networks☆13Jan 31, 2024Updated 2 years ago
- Google Collab Notebooks for the UNIL Spring 2022 course on ML for Earth and Environmental Sciences☆14Aug 18, 2022Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 3 years ago
- Official implementation of the AAMAS 2024 paper Measruing Policy Distance for MARL.☆11Mar 4, 2026Updated 8 months ago