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. 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 5 years ago
- A collection of meta-learning algorithms in Jax☆25Sep 3, 2022Updated 3 years ago
- flexible meta-learning in jax☆16Oct 19, 2023Updated 2 years ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Mar 4, 2022Updated 4 years ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 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.
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 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 4 years ago
- Implementation for Non-stationary Spectral Kernels (NIPS 2017)☆20Jan 4, 2020Updated 6 years ago
- Adaptable Agent Populations via a Generative Model of Policies☆12Oct 14, 2021Updated 4 years ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 years ago
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 years ago
- 3D geoms for plotnine (grammar of graphics in Python)☆13Aug 5, 2022Updated 3 years ago
- A framework for composing Neural Processes in Julia☆76May 3, 2021Updated 5 years ago
- Maximum Entropy-Regularized Multi-Goal Reinforcement Learning (ICML 2019)☆24May 30, 2019Updated 6 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- 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
- Code for utilising VAE as means of doing exact MCMC inference in complex high-dimensional space☆14Jun 20, 2023Updated 2 years ago
- Code for Deep Structured Mixtures of Gaussian Processes (DSMGPs)☆11Jan 27, 2022Updated 4 years ago
- Mental state inference from observable behavior☆15Dec 3, 2021Updated 4 years ago
- ☆17Oct 19, 2020Updated 5 years ago
- Normalizing Flows using JAX☆87Nov 30, 2023Updated 2 years ago
- Multiple dispatch over abstract array types in JAX.☆141Updated this week
- A pathway and collection of resources to learning Jax from beginning to advance.☆11Jan 2, 2021Updated 5 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
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Prism themes for plotnine, inspired by ggprism.☆17Aug 13, 2025Updated 8 months ago
- ☆11Apr 2, 2022Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 3 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
- This repo is the implementation of paper ''SHAQ: Incorporating Shapley Value Theory into Multi-Agent Q-Learning''.☆51Dec 4, 2023Updated 2 years ago
- Probabilistic deep learning using JAX☆15Feb 8, 2025Updated last year
- PSYCH 291: Causal Cognition (https://tobiasgerstenberg.github.io/causal_cognition/)☆13May 23, 2019Updated 6 years ago
- Cooperation and Fairness in Multi-Agent Reinforcement Learning☆16Aug 6, 2025Updated 9 months ago
- ☆12Mar 2, 2021Updated 5 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆18Nov 24, 2025Updated 5 months ago
- Procgen2: A community maintained fork of procgen☆12Aug 25, 2022Updated 3 years ago
- R&D Python package for processing ICON data with xarray☆10Jul 26, 2023Updated 2 years ago
- Repository for the HPC & Data Science Summer Institute 2024☆12Mar 7, 2025Updated last year
- Goal-conditioned reinforcement learning like 🔥☆15Feb 3, 2024Updated 2 years ago
- Code for the paper "FinRLlama: A Solution to LLM-Engineered Signals Challenge at FinRL Contest 2024"☆13Feb 14, 2025Updated last year
- run a script as it will be run from cron☆15Aug 18, 2022Updated 3 years ago