Pinned Repositories
clip-guided-diffusion
CLIP Guided Diffusion
deep_dream
A parallel implementation of the Deep Dream image processing algorithm which is able to process arbitrarily large images.
guided-diffusion
jax-wavelets
The 2D discrete wavelet transform for JAX
k-diffusion
Karras et al. (2022) diffusion models for PyTorch
simulacra-aesthetic-models
style-transfer-pytorch
Neural style transfer in PyTorch.
style_transfer
Data-parallel image stylization using Caffe.
v-diffusion-jax
v objective diffusion inference code for JAX.
v-diffusion-pytorch
v objective diffusion inference code for PyTorch.
crowsonkb's Repositories
crowsonkb/k-diffusion
Karras et al. (2022) diffusion models for PyTorch
crowsonkb/v-diffusion-pytorch
v objective diffusion inference code for PyTorch.
crowsonkb/style-transfer-pytorch
Neural style transfer in PyTorch.
crowsonkb/v-diffusion-jax
v objective diffusion inference code for JAX.
crowsonkb/simulacra-aesthetic-models
crowsonkb/guided-diffusion
crowsonkb/consistency-models
A JAX implementation of the continuous time formulation of Consistency Models
crowsonkb/cloob-training
CLOOB training (JAX) and inference (JAX and PyTorch)
crowsonkb/LDLM
Latent Diffusion Language Models
crowsonkb/latent-diffusion
High-Resolution Image Synthesis with Latent Diffusion Models
crowsonkb/clip-guided-diffusion
CLIP Guided Diffusion
crowsonkb/esgd
ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.
crowsonkb/mdmm
The Modified Differential Multiplier Method (MDMM) for PyTorch
crowsonkb/deep-image-prior
Image restoration with neural networks but without learning.
crowsonkb/jax-wavelets
The 2D discrete wavelet transform for JAX
crowsonkb/dice-mc
DiCE: The Infinitely Differentiable Monte-Carlo Estimator
crowsonkb/mdmm-jax
Gradient-based constrained optimization for JAX
crowsonkb/torch-dist-utils
Utilities for PyTorch distributed
crowsonkb/cond_transformer_2
A CLIP conditioned Decision Transformer.
crowsonkb/glide-text2im
GLIDE: a diffusion-based text-conditional image synthesis model
crowsonkb/pytorch-caffe-models
The original weights of some Caffe models, ported to PyTorch.
crowsonkb/rope-flax
Rotary Position Embedding for Flax
crowsonkb/simple-diffusion-model
Pedagogical codebase for a simplified score-based generative model design, with training loop
crowsonkb/stable-diffusion
crowsonkb/kat-wip
works in progress
crowsonkb/torch-fidelity
High-fidelity performance metrics for generative models in PyTorch
crowsonkb/exploring-contrastive-topology
crowsonkb/flash-attention
Fast and memory-efficient exact attention
crowsonkb/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
crowsonkb/lpips-jax