Alternatives and similar repositories for hackable_diffusion

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

• mgrankin / minGPT
  minGPT in JAX
  ☆48Updated 3 years ago
• Artur-Galstyan / statedict2pytree
  ☆44Updated last month
• mle-infrastructure / mle-scheduler
  Lightweight Cluster/Cloud VM Job Management 🚀
  ☆42Updated last year
• srush / torch-queue
  ☆71Updated 2 years ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated last month
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆118Updated 3 years ago
• aks2203 / deep-thinking
  A centralized place for deep thinking code and experiments
  ☆88Updated 2 years ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆88Updated last year
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆114Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆188Updated last week
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆33Updated 2 years ago
• Ping-C / optimizer
  This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…
  ☆40Updated 2 years ago
• AllanYangZhou / universal_neural_functional
  ☆53Updated last year
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆54Updated 2 years ago
• Felix-Petersen / algovision
  Differentiable Algorithms and Algorithmic Supervision.
  ☆116Updated 2 years ago
• louiskirsch / vsml-neurips2021
  Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905
  ☆33Updated 3 years ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆143Updated last year
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆81Updated 3 years ago
• 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
• google-deepmind / jmp
  JMP is a Mixed Precision library for JAX.
  ☆211Updated 10 months ago
• GallagherCommaJack / modulax
  ☆17Updated last year
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆191Updated last year
• HomebrewML / revlib
  Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
  ☆131Updated 3 years ago
• davisyoshida / haiku-mup
  A port of muP to JAX/Haiku
  ☆25Updated 3 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"
  ☆60Updated 3 years ago
• cgarciae / einop
  ☆60Updated 3 years ago
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆24Updated 3 years ago
• ejmichaud / grokking-squared
  ☆27Updated 2 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆38Updated 3 years ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆40Updated last year