crowsonkb / mdmm-jax

Related projects ⓘ

Alternatives and complementary repositories for mdmm-jax

• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• google-deepmind / spectral_ssm
  ☆28Updated 7 months ago
• brentyi / minGPT-flax
  GPT implementation in Flax
  ☆18Updated 2 years ago
• FLAIROx / jaxirl
  Contains JAX implementation of algorithms for inverse reinforcement learning
  ☆63Updated 3 months ago
• enajx / HyperNCA
  ☆36Updated 2 years ago
• brentyi / jax_dataclasses
  Pytrees + dataclasses ❤️
  ☆64Updated 4 months ago
• COR-OPT / SuperPolyak.py
  Pytorch implementation of SuperPolyak subgradient method.
  ☆43Updated 2 years ago
• juliuskunze / cwvae-jax
  Clockwork VAEs in JAX/Flax
  ☆32Updated 3 years ago
• danijar / elements
  Building blocks for productive research
  ☆45Updated last week
• smonsays / metax
  flexible meta-learning in jax
  ☆12Updated last year
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆40Updated 2 years ago
• danijar / ninjax
  General Modules for JAX
  ☆58Updated 3 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆77Updated 3 months ago
• keraJLi / synthetic-gymnax
  ☆17Updated 5 months ago
• ssokota / mec
  Code for minimum-entropy coupling.
  ☆30Updated 4 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders in JAX.
  ☆59Updated last month
• GallagherCommaJack / modulax
  ☆16Updated 2 months ago
• DramaCow / jaxued
  ☆63Updated 3 months ago
• icaros-usc / dqd
  A python implementation of differentiable quality diversity.
  ☆48Updated 3 years ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated 3 weeks ago
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆96Updated last year
• Reytuag / transformerXL_PPO_JAX
  ☆65Updated 2 weeks ago
• neuroevobench / neuroevobench
  Neuroevolution Benchmark in JAX 🦕
  ☆37Updated last year
• smearle / autoverse
  Generative cellular automaton-like learning environments for RL.
  ☆19Updated last month
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆37Updated 7 months ago
• google-deepmind / dm_hamiltonian_dynamics_suite
  ☆34Updated 3 years ago
• instadeepai / fastpbrl
  Vectorization techniques for fast population-based training.
  ☆54Updated 2 years ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX
  ☆71Updated this week
• twitter-research / hyperbolic-rl
  ☆56Updated 2 years ago