Alternatives and similar repositories for mdmm-jax:

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

• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 3 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆41Updated 2 months ago
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated last year
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆90Updated last year
• ASEM000 / pytreeclass
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆43Updated 2 weeks ago
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆79Updated 2 years ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders in JAX.
  ☆72Updated 3 months ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆24Updated last month
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆63Updated 6 months ago
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆36Updated 5 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆83Updated 5 months ago
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆44Updated last week
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆34Updated 3 years ago
• bb515 / diffusionjax
  diffusionjax is a simple and accessible diffusion models package in JAX
  ☆40Updated last week
• enajx / HyperNCA
  ☆37Updated 2 years ago
• google-deepmind / spectral_ssm
  ☆31Updated 9 months ago
• GallagherCommaJack / modulax
  ☆15Updated 5 months ago
• cgarciae / simple_flow_matching
  ☆14Updated last month
• crowsonkb / mdmm
  The Modified Differential Multiplier Method (MDMM) for PyTorch
  ☆55Updated 3 years ago
• icaros-usc / dqd
  A python implementation of differentiable quality diversity.
  ☆48Updated 3 years ago
• Kaixhin / GUDRL
  Generalised UDRL
  ☆37Updated 2 years ago
• DramaCow / jaxued
  ☆67Updated 5 months ago
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• google-deepmind / tf2jax
  ☆111Updated 3 weeks ago
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• google-deepmind / annealed_flow_transport
  ☆49Updated last year
• danijar / elements
  Building blocks for productive research
  ☆47Updated this week
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• cambridge-mlg / neural_diffusion_processes
  ☆13Updated last year