google / hackable_diffusionLinks
☆25Updated last month
Alternatives and similar repositories for hackable_diffusion
Users that are interested in hackable_diffusion are comparing it to the libraries listed below
Sorting:
- minGPT in JAX☆48Updated 3 years ago
- ☆44Updated last month
- Lightweight Cluster/Cloud VM Job Management 🚀☆42Updated last year
- ☆71Updated 2 years ago
- A simple library for scaling up JAX programs☆144Updated last month
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆118Updated 3 years ago
- A centralized place for deep thinking code and experiments☆88Updated 2 years ago
- A functional training loops library for JAX☆88Updated last year
- Running Jax in PyTorch Lightning☆114Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆188Updated last week
- Open source code for EigenGame.☆33Updated 2 years ago
- 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
- ☆53Updated last year
- Scaling scaling laws with board games.☆54Updated 2 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆116Updated 2 years ago
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆33Updated 3 years ago
- LoRA for arbitrary JAX models and functions☆143Updated last year
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆81Updated 3 years ago
- Minimalistic, hackable PyTorch implementation of SimSiam in ~400 lines. Achieves good performance on ImageNet with ResNet50. Features dis…☆21Updated last year
- JMP is a Mixed Precision library for JAX.☆211Updated 10 months ago
- ☆17Updated last year
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆191Updated last year
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆131Updated 3 years ago
- A port of muP to JAX/Haiku☆25Updated 3 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 3 years ago
- ☆60Updated 3 years ago
- A simple hypernetwork implementation in jax using haiku.☆24Updated 3 years ago
- ☆27Updated 2 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆38Updated 3 years ago
- Image augmentation library for Jax☆40Updated last year