Alternatives and similar repositories for stable-diffusion-jax

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

• borisdayma / clip-jax

  View on GitHub
  Train vision models using JAX and 🤗 transformers
  ☆100Dec 14, 2025Updated last month
• patil-suraj / vqgan-jax

  View on GitHub
  JAX implementation of VQGAN
  ☆91Jul 9, 2022Updated 3 years ago
• yiyixuxu / denoising-diffusion-flax

  View on GitHub
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆156Nov 1, 2022Updated 3 years ago
• tgisaturday / dalle-lightning

  View on GitHub
  Refactoring dalle-pytorch and taming-transformers for TPU VM
  ☆60Aug 30, 2021Updated 4 years ago
• yardenas / jax-dreamer

  View on GitHub
  Dreamer on JAX
  ☆16Jan 19, 2022Updated 4 years ago
• patil-suraj / vit-vqgan

  View on GitHub
  JAX implementation ViT-VQGAN
  ☆82Sep 21, 2022Updated 3 years ago
• jiyounglee-0523 / FourierDecoder

  View on GitHub
  Official repository for Fourier model that can generate periodic signals
  ☆10Mar 10, 2022Updated 3 years ago
• nshepperd / jax-guided-diffusion

  View on GitHub
  jax version of clip guided diffusion scripts
  ☆90Jan 11, 2024Updated 2 years ago
• rwightman / efficientnet-jax

  View on GitHub
  EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
  ☆129Jan 4, 2024Updated 2 years ago
• JesseFarebro / flax-mup

  View on GitHub
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Dec 27, 2023Updated 2 years ago
• matthias-wright / flaxmodels

  View on GitHub
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆265Mar 21, 2025Updated 10 months ago
• Dango233 / PrincessGenerator

  View on GitHub
  Guided diffusion
  ☆11May 22, 2022Updated 3 years ago
• n2cholas / progan-flax

  View on GitHub
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12May 24, 2021Updated 4 years ago
• lucidrains / CLAP

  View on GitHub
  Contrastive Language-Audio Pretraining
  ☆15May 18, 2021Updated 4 years ago
• SonicCodes / subcloning

  View on GitHub
  implementation of https://arxiv.org/pdf/2312.09299
  ☆21Jul 3, 2024Updated last year
• Jack000 / guided-diffusion

  View on GitHub
  ☆28Dec 16, 2021Updated 4 years ago
• kingoflolz / CLIP_JAX

  View on GitHub
  Contrastive Language-Image Pretraining
  ☆144Sep 6, 2022Updated 3 years ago
• crowsonkb / latent-diffusion

  View on GitHub
  High-Resolution Image Synthesis with Latent Diffusion Models
  ☆60Apr 17, 2022Updated 3 years ago
• pcuenca / lpips-j

  View on GitHub
  Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.
  ☆17Aug 1, 2022Updated 3 years ago
• Sea-Snell / JAXSeq

  View on GitHub
  Train very large language models in Jax.
  ☆210Oct 21, 2023Updated 2 years ago
• Jack000 / glid-3

  View on GitHub
  combination of OpenAI GLIDE and Latent Diffusion
  ☆136Apr 7, 2022Updated 3 years ago
• HomebrewML / Olmax

  View on GitHub
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆51Jan 20, 2024Updated 2 years ago
• dmarx / Multi-Modal-Comparators

  View on GitHub
  Unified API to facilitate usage of pre-trained "perceptor" models, a la CLIP
  ☆39Nov 26, 2022Updated 3 years ago
• dome272 / Paella

  View on GitHub
  Official Implementation of Paella https://arxiv.org/abs/2211.07292v2
  ☆748Oct 4, 2023Updated 2 years ago
• ayaka14732 / llama-2-jax

  View on GitHub
  JAX implementation of the Llama 2 model
  ☆216Feb 2, 2024Updated 2 years ago
• vballoli / vit-flax

  View on GitHub
  Implementation of Vision Transformers in Flax
  ☆18Oct 12, 2020Updated 5 years ago
• lucidrains / local-attention-flax

  View on GitHub
  Local Attention - Flax module for Jax
  ☆22May 26, 2021Updated 4 years ago
• conceptofmind / vit-flax

  View on GitHub
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Aug 30, 2022Updated 3 years ago
• duskvirkus / colab-notebooks

  View on GitHub
  A collection of my machine learning notebooks to run on google colab. Mostly ml art.
  ☆20Jun 10, 2022Updated 3 years ago
• SelfishGene / visual_taste_approximator

  View on GitHub
  Visual Taste Approximator (VTA) is a very simple tool that helps anyone create an automatic replica of themselves that can approximate th…
  ☆39Sep 11, 2022Updated 3 years ago
• jaeseokbyun / GRIT-VLP

  View on GitHub
  This is an official implementation of GRIT-VLP
  ☆20Aug 8, 2022Updated 3 years ago
• LAION-AI / laion50BU

  View on GitHub
  Un-*** 50 billions multimodality dataset
  ☆23Sep 14, 2022Updated 3 years ago
• kuprel / min-dalle-flax

  View on GitHub
  This contains the Flax model of min(DALL·E) and code for converting it to PyTorch
  ☆45Jul 21, 2022Updated 3 years ago
• Jack000 / ru-dalle

  View on GitHub
  Generate images from texts. In Russian
  ☆19Dec 13, 2021Updated 4 years ago
• lucidrains / discrete-key-value-bottleneck-pytorch

  View on GitHub
  Implementation of Discrete Key / Value Bottleneck, in Pytorch
  ☆88Jul 9, 2023Updated 2 years ago
• leszekhanusz / diffusion-ui-backend

  View on GitHub
  Backend for the diffusion-ui frontend
  ☆25Feb 17, 2024Updated last year
• AranKomat / Metroplex

  View on GitHub
  ☆21Mar 15, 2023Updated 2 years ago
• cgarciae / ciclo

  View on GitHub
  A functional training loops library for JAX
  ☆88Feb 13, 2024Updated 2 years ago
• ethansmith2000 / StableTwoUnet

  View on GitHub
  ☆26Jun 5, 2024Updated last year