Alternatives and similar repositories for hackable_diffusion

Users that are interested in hackable_diffusion are comparing it to the libraries listed below

• Artur-Galstyan / statedict2pytree
  ☆45Updated this week
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• mgrankin / minGPT
  minGPT in JAX
  ☆48Updated 4 years ago
• srush / torch-queue
  ☆71Updated 2 years ago
• GallagherCommaJack / modulax
  ☆18Updated last year
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆34Updated 2 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆119Updated 3 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆119Updated last year
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• lucas-maes / nano-simsiam
  Minimalistic, hackable PyTorch implementation of SimSiam in ~400 lines. Achieves good performance on ImageNet with ResNet50. Features dis…
  ☆21Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆190Updated 3 weeks ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆41Updated last year
• cgarciae / einop
  ☆60Updated 3 years ago
• mle-infrastructure / mle-scheduler
  Lightweight Cluster/Cloud VM Job Management 🚀
  ☆42Updated last year
• nestordemeure / jochastic
  A JAX implementation of stochastic addition.
  ☆14Updated 3 years ago
• google-deepmind / neural_testbed
  ☆195Updated 3 weeks ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆39Updated 3 years ago
• joschu / jax-exp
  ☆24Updated 7 years ago
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆53Updated 2 years ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆259Updated 2 years ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆61Updated 3 years ago
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆37Updated last year
• cgarciae / treex
  A Pytree Module system for Deep Learning in JAX
  ☆214Updated 2 years ago
• google-deepmind / einshape
  ☆109Updated last year
• MichaelArbel / GeneralizedEBM
  ☆54Updated last year
• Felix-Petersen / algovision
  Differentiable Algorithms and Algorithmic Supervision.
  ☆116Updated 2 years ago
• mila-iqia / milatools
  Tools to connect to and interact with the Mila cluster
  ☆79Updated last month
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆211Updated last year
• google-deepmind / tf2jax
  ☆120Updated last week
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆45Updated 6 months ago