jax-comln

Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)

MIT License

Stars
27
View Code on GitHub View on X
Ecosystems: Python

Statistics for this project are still being loaded, please check back later.

Related Projects
nn-tutorials

27 Feb 2017 6
metax

A collection of extensions for meta-learning in JAX

23 Oct 2020 8
CLsurvey

Continual Hyperparameter Selection Framework. Compares 11 state-of-the-art Lifelong Learning meth...

06 Apr 2020 192
pytorch-maml-rl

Reinforcement Learning with Model-Agnostic Meta-Learning in Pytorch

28 Apr 2018 809
Lightweighting_Cookbook

This project attempts to build neural network training and lightweighting cookbook including thre...

25 May 2022 18
equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

29 Jul 2021 1,846
pytorch-maml

An Implementation of Model-Agnostic Meta-Learning in PyTorch with Torchmeta

14 Dec 2018 226
jax-meta-learning

A collection of meta-learning algorithms in Jax

31 Mar 2021 23
lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

09 May 2023 345
meta-blocks

A modular toolbox for meta-learning research with a focus on speed and reproducibility.

24 Feb 2020 123
mlp-gpt-jax

A GPT, made only of MLPs, in Jax

21 May 2021 55
rlax

18 Feb 2020 1,193
optax

Optax is a gradient processing and optimization library for JAX.

12 Jun 2020 1,505
byol-pytorch

Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in ...

16 Jun 2020 1,687
jax-dag-gflownet

Code for "Bayesian Structure Learning with Generative Flow Networks"

21 Mar 2022 36