Alternatives and similar repositories for jax-meta-learning

Users that are interested in jax-meta-learning are comparing it to the libraries listed below

• nestordemeure / jochastic

  View on GitHub
  A JAX implementation of stochastic addition.
  ☆14Aug 15, 2022Updated 3 years ago
• alexub / jax-meta-learning

  View on GitHub
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆11Oct 6, 2020Updated 5 years ago
• SforAiDl / jeta

  View on GitHub
  A Jax based meta learning library
  ☆16Sep 15, 2022Updated 3 years ago
• gcucurull / maml_flax

  View on GitHub
  Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.
  ☆21Sep 18, 2020Updated 5 years ago
• smonsays / metax

  View on GitHub
  flexible meta-learning in jax
  ☆16Oct 19, 2023Updated 2 years ago
• tristandeleu / gfn-maxent-rl

  View on GitHub
  Comparison between GFlowNets & Maximum Entropy RL
  ☆19Feb 19, 2024Updated last year
• brentyi / jax_dataclasses

  View on GitHub
  Pytrees + dataclasses ❤️
  ☆74Dec 19, 2025Updated last month
• Aladoro / Stabilizing-Off-Policy-RL

  View on GitHub
  ☆17Aug 3, 2022Updated 3 years ago
• JesseFarebro / distributional-sr

  View on GitHub
  Official implementation of the δ-model presented in the ICML 2024 paper "A Distributional Analogue to the Successor Representation".
  ☆25Nov 8, 2024Updated last year
• mgrankin / minGPT

  View on GitHub
  minGPT in JAX
  ☆48Jan 10, 2022Updated 4 years ago
• henry-prior / jax-rl

  View on GitHub
  JAX implementations of core Deep RL algorithms
  ☆83May 2, 2022Updated 3 years ago
• cgarciae / treeo

  View on GitHub
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Feb 26, 2023Updated 2 years ago
• NicolasZucchet / minimal-LRU

  View on GitHub
  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)
  ☆62Sep 3, 2025Updated 5 months ago
• 8bitmp3 / JAX-Flax-Tutorial-Image-Classification-with-Linen

  View on GitHub
  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
• young-geng / mintext

  View on GitHub
  Minimal but scalable implementation of large language models in JAX
  ☆35Nov 28, 2025Updated 2 months ago
• alexlioralexli / rllab-finetuning

  View on GitHub
  ☆30Jan 19, 2023Updated 3 years ago
• ikostrikov / jaxrl

  View on GitHub
  JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
  ☆750Oct 26, 2022Updated 3 years ago
• eml-eda / pit

  View on GitHub
  Lightweight Neural Architecture Search for Temporal Convolutional Networks at the Edge
  ☆10Mar 6, 2023Updated 2 years ago
• jliang993 / A3DMM

  View on GitHub
  Source code for paper "Trajectory of Alternating Direction Method of Multipliers and Adaptive Acceleration" of NeurIPS 2019
  ☆10Jan 25, 2024Updated 2 years ago
• parasdahal / deepnotes

  View on GitHub
  Source code for my blog DeepNotes
  ☆15May 4, 2023Updated 2 years ago
• yardenas / actsafe

  View on GitHub
  Scaling safe exploration to vision control
  ☆14Feb 19, 2025Updated 11 months ago
• instadeepai / flashbax

  View on GitHub
  ⚡ Flashbax: Accelerated Replay Buffers in JAX
  ☆272Sep 22, 2025Updated 4 months ago
• pylat / adaptive-proximal-algorithms

  View on GitHub
  A Julia package for adaptive proximal gradient and primal-dual algorithms
  ☆11Jan 18, 2024Updated 2 years ago
• josephsalmon / HAX712X

  View on GitHub
  Software dev. for data science (Python)
  ☆15May 1, 2025Updated 9 months ago
• hoburg / pyxfoil

  View on GitHub
  Python tools for running xfoil many times
  ☆10Sep 10, 2016Updated 9 years ago
• vitchyr / torch-rl

  View on GitHub
  A reinforcement learning package implemented in Torch
  ☆11Jan 24, 2016Updated 10 years ago
• EdanToledo / Stoix

  View on GitHub
  🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL
  ☆390Oct 29, 2025Updated 3 months ago
• yardenas / la-mbda

  View on GitHub
  LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization
  ☆38Jan 16, 2023Updated 3 years ago
• hr0nix / dejax

  View on GitHub
  Accelerated replay buffers in JAX
  ☆46Sep 17, 2022Updated 3 years ago
• danielkalen / foreach-cli

  View on GitHub
  CLI util to execute a command for each file matching a glob
  ☆12Nov 7, 2023Updated 2 years ago
• SunHaozhe / OmniPrint

  View on GitHub
  ☆11Oct 31, 2021Updated 4 years ago
• jiasenlu / vit-vqgan-jax

  View on GitHub
  Jax implementation of VIT-VQGAN
  ☆10Jan 25, 2024Updated 2 years ago
• TheoGuyard / El0ps

  View on GitHub
  El0ps: An Exact L0-Problem Solver
  ☆13Jan 6, 2026Updated last month
• timothyf / baseball-data-lab

  View on GitHub
  A Python application and library that generates comprehensive advanced stat summary sheets for MLB players, customizable by year, providi…
  ☆14Sep 12, 2025Updated 5 months ago
• antoninschrab / mmdfuse

  View on GitHub
  MMD-FUSE package implementing the MMD-FUSE test proposed in MMD-FUSE: Learning and Combining Kernels for Two-Sample Testing Without Data …
  ☆11May 31, 2024Updated last year
• peterwestuw / GPT2ForwardBackward

  View on GitHub
  Code for running forward and backward versions of GPT2
  ☆10Nov 20, 2021Updated 4 years ago
• instadeepai / outer-value-function-meta-rl

  View on GitHub
  Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Function
  ☆13Nov 22, 2022Updated 3 years ago
• mattjj / ode-diff-notes

  View on GitHub
  ☆12Oct 8, 2020Updated 5 years ago
• lucataco / cog-ssd-1b

  View on GitHub
  Attempt at cog wrapper for segmind/SSD-1B
  ☆10Dec 11, 2023Updated 2 years ago