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.
  ☆25Updated last month
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆79Updated 2 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆62Updated 8 months ago
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆20Updated 9 months ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆26Updated 9 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆86Updated 7 months ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆90Updated last year
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆34Updated 3 years ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• DramaCow / jaxued
  ☆75Updated 2 weeks ago
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆22Updated 2 years ago
• nikihowe / myriad
  Myriad is a real-world testbed that aims to bridge trajectory optimization and deep learning.
  ☆66Updated last year
• GallagherCommaJack / modulax
  ☆17Updated 7 months ago
• danijar / ninjax
  General Modules for JAX
  ☆64Updated last month
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆41Updated 5 months ago
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 2 years ago
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆45Updated 2 months ago
• FLAIROx / jaxirl
  Contains JAX implementation of algorithms for inverse reinforcement learning
  ☆71Updated 7 months ago
• EmptyJackson / groove
  Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…
  ☆25Updated 9 months ago
• flowersteam / vivarium
  Multi-agent simulator in Jax for research and teaching in AI & ALife
  ☆27Updated 3 weeks ago
• vsimkus / torch-reparametrised-mixture-distribution
  PyTorch implementation of the mixture distribution family with implicit reparametrisation gradients.
  ☆19Updated last year
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆77Updated 5 months ago
• enajx / HyperNCA
  ☆37Updated 2 years ago
• qsh-zh / pis
  [ICLR 2022] Path integral sampler
  ☆45Updated last year
• MinRegret / deluca
  Performant, differentiable reinforcement learning
  ☆25Updated last year
• AlexGoldie / rl-learned-optimization
  Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"
  ☆22Updated 4 months ago