Kale Champagnie's repositories
axial-attention
An implementation of axial attention in PyTorch.
datamodules
Data modules for PyTorch 🔥.
gaussian-mixure-nll-loss
An implementation of Gaussian mixture NLL loss for continuous density estimation.
generative-models
Generative Models by Stability AI
golf-flax
A golfed version of the Flax module API.
half-blood-prince
Harry Potter and the Half Blood Prince in plain text form.
minimal-transformer
A minimal decoder-only transformer implemented in under 50 lines of PyTorch.
minimal-transformer-tensorflow
A minimal transformer implemented in TensorFlow.
stable-diffusion
An implementation of Stable Diffusion in PyTorch (WIP).
stablediffusion
High-Resolution Image Synthesis with Latent Diffusion Models
temporal-language-model
An implementation of Temporal Language Model from "Temporal Attention for Language Models", in PyTorch.
vq-vae
An implementation of VQ-VAE in PyTorch.
weights-reset
An implementation of Weights Reset from "The Weights Reset Technique for Deep Neural Networks Implicit Regularization".