Alternatives and similar repositories for mdmm-jax:

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

• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆24Updated 2 weeks ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆63Updated 7 months ago
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆90Updated last year
• google-deepmind / spectral_ssm
  ☆31Updated 10 months ago
• enajx / HyperNCA
  ☆37Updated 2 years ago
• danijar / ninjax
  General Modules for JAX
  ☆64Updated this week
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆75Updated 4 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆85Updated 6 months ago
• Z-T-WANG / LaProp-Optimizer
  Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"
  ☆27Updated 4 years ago
• DramaCow / jaxued
  ☆72Updated 6 months ago
• Michael-Beukman / RobocupGym
  Reinforcement Learning inside a 3D soccer simulation
  ☆25Updated 5 months ago
• GallagherCommaJack / modulax
  ☆16Updated 6 months ago
• FLAIROx / jaxirl
  Contains JAX implementation of algorithms for inverse reinforcement learning
  ☆69Updated 6 months ago
• ivy-llc / gym
  Fully differentiable RL environments, written in Ivy.
  ☆64Updated last year
• crowsonkb / mdmm
  The Modified Differential Multiplier Method (MDMM) for PyTorch
  ☆57Updated 3 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• alexub / jax-meta-learning
  Simple, extensible implementations of some meta-learning algorithms in Jax
  ☆9Updated 4 years ago
• akbir / deq-jax
  [NeurIPS'19] Deep Equilibrium Models Jax Implementation
  ☆39Updated 4 years ago
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated last year
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆33Updated last year
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated last month
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆43Updated last month
• google-deepmind / einshape
  ☆101Updated 8 months ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆56Updated 2 years ago
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆73Updated 3 weeks ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆27Updated 7 months ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆41Updated 4 months ago