Alternatives and similar repositories for stable-diffusion-jax

Users that are interested in stable-diffusion-jax are comparing it to the libraries listed below

• crowsonkb / consistency-models
  A JAX implementation of the continuous time formulation of Consistency Models
  ☆84Updated 2 years ago
• JD-P / cloob-latent-diffusion
  CLOOB Conditioned Latent Diffusion training and inference code
  ☆112Updated 3 years ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆97Updated last month
• afiaka87 / latent-diffusion-deepspeed
  Finetune the 1.4B latent diffusion text2img-large checkpoint from CompVis using deepspeed. (work-in-progress)
  ☆36Updated 3 years ago
• crowsonkb / cloob-training
  CLOOB training (JAX) and inference (JAX and PyTorch)
  ☆72Updated 3 years ago
• cloneofsimo / sd-various-ideas
  ☆56Updated 2 years ago
• kingoflolz / CLIP_JAX
  Contrastive Language-Image Pretraining
  ☆141Updated 2 years ago
• HazyResearch / diffusers
  🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch
  ☆50Updated 2 years ago
• patil-suraj / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆83Updated 2 years ago
• crowsonkb / latent-diffusion
  High-Resolution Image Synthesis with Latent Diffusion Models
  ☆61Updated 3 years ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated 2 years ago
• HomebrewML / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆42Updated last year
• crowsonkb / v-diffusion-jax
  v objective diffusion inference code for JAX.
  ☆214Updated 3 years ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆42Updated 2 years ago
• pbaylies / Augmented_CLIP
  Training simple models to predict CLIP image embeddings from text embeddings, and vice versa.
  ☆60Updated 3 years ago
• crowsonkb / esgd
  ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.
  ☆56Updated 2 years ago
• patil-suraj / vqgan-jax
  JAX implementation of VQGAN
  ☆93Updated 2 years ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆136Updated last year
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆146Updated 2 years ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 10 months ago
• crowsonkb / LDLM
  Latent Diffusion Language Models
  ☆67Updated last year
• Birch-san / diffusers-play
  Repository with which to explore k-diffusion and diffusers, and within which changes to said packages may be tested.
  ☆53Updated last year
• pabloppp / Arroz-Con-Cosas
  Experimental LDM uses of Paella's architecture
  ☆34Updated 2 years ago
• cloneofsimo / ezmup
  Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam
  ☆77Updated 10 months ago
• dmarx / Multi-Modal-Comparators
  Unified API to facilitate usage of pre-trained "perceptor" models, a la CLIP
  ☆38Updated 2 years ago
• cloneofsimo / karras-power-ema-tutorial
  ☆51Updated last year
• cloneofsimo / zeroshampoo
  ☆33Updated 8 months ago
• nshepperd / jax-guided-diffusion
  jax version of clip guided diffusion scripts
  ☆90Updated last year
• pcuenca / lpips-j
  Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.
  ☆15Updated 2 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆44Updated last year