patil-suraj/stable-diffusion-jax external links

---

GitHub - View repository

GitHub.dev - Open in online code editor

Star history - Growth chart over time

deps.dev - Dependencies and security vulnerabilities

Gitingest - Export source code for LLM usage

Repomix - Export source code for LLM usage

uithub - Export source code for LLM usage

Close

patil-suraj/stable-diffusion-jax

Readme badge preview -

If you own this repo, copy the snippet below and add it to your README.md

---

Copy

[![RelatedRepos](https://img.shields.io/badge/related-repos-yellow)](https://relatedrepos.com/gh/patil-suraj/stable-diffusion-jax)

Close

patil-suraj / stable-diffusion-jaxView on GitHubMore

• External links
• Readme badge

☆91Sep 19, 2022Updated 3 years ago

Alternatives and similar repositories for stable-diffusion-jax

Users that are interested in stable-diffusion-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

• borisdayma / clip-jax

  View on GitHub
  Train vision models using JAX and 🤗 transformers
  ☆102Dec 14, 2025Updated 4 months ago
• 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
  ☆161Nov 1, 2022Updated 3 years ago
• rwightman / efficientnet-jax

  View on GitHub
  EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax
  ☆130Jan 4, 2024Updated 2 years ago
• yardenas / jax-dreamer

  View on GitHub
  Dreamer on JAX
  ☆16Jan 19, 2022Updated 4 years ago
• Virtual machines for every use case on DigitalOcean • Ad
  Get dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
• tgisaturday / dalle-lightning

  View on GitHub
  Refactoring dalle-pytorch and taming-transformers for TPU VM
  ☆60Aug 30, 2021Updated 4 years ago
• patil-suraj / vit-vqgan

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

  View on GitHub
  Contrastive Language-Audio Pretraining
  ☆15May 18, 2021Updated 4 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.
  ☆264Mar 21, 2025Updated last year
• crowsonkb / latent-diffusion

  View on GitHub
  High-Resolution Image Synthesis with Latent Diffusion Models
  ☆60Apr 17, 2022Updated 4 years ago
• jiyounglee-0523 / FourierDecoder

  View on GitHub
  Official repository for Fourier model that can generate periodic signals
  ☆10Mar 10, 2022Updated 4 years ago
• DarshanDeshpande / jax-models

  View on GitHub
  Unofficial JAX implementations of deep learning research papers
  ☆162Jun 25, 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
• GPUs on demand by Runpod - Special Offer Available • Ad
  Run AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
• 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
• ayaka14732 / llama-2-jax

  View on GitHub
  JAX implementation of the Llama 2 model
  ☆216Feb 2, 2024Updated 2 years ago
• nestordemeure / jochastic

  View on GitHub
  A JAX implementation of stochastic addition.
  ☆14Aug 15, 2022Updated 3 years ago
• nshepperd / jax-guided-diffusion

  View on GitHub
  jax version of clip guided diffusion scripts
  ☆91Jan 11, 2024Updated 2 years ago
• cgarciae / treex

  View on GitHub
  A Pytree Module system for Deep Learning in JAX
  ☆212Feb 26, 2023Updated 3 years ago
• kingoflolz / CLIP_JAX

  View on GitHub
  Contrastive Language-Image Pretraining
  ☆146Sep 6, 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
• HomebrewML / Olmax

  View on GitHub
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆51Jan 20, 2024Updated 2 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
• 1-Click AI Models by DigitalOcean Gradient • Ad
  Deploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
• Dango233 / PrincessGenerator

  View on GitHub
  Guided diffusion
  ☆11May 22, 2022Updated 3 years ago
• cgarciae / ciclo

  View on GitHub
  A functional training loops library for JAX
  ☆88Feb 13, 2024Updated 2 years ago
• julien-c / trainer-proposal

  View on GitHub
  ☆13Mar 27, 2020Updated 6 years ago
• google / flaxformer

  View on GitHub
  ☆368Apr 12, 2024Updated 2 years ago
• ASEM000 / pytreeclass

  View on GitHub
  Visualize, create, and operate on pytrees in the most intuitive way possible.
  ☆46Jan 11, 2025Updated last year
• leszekhanusz / diffusion-ui-backend

  View on GitHub
  Backend for the diffusion-ui frontend
  ☆25Feb 17, 2024Updated 2 years ago
• Jack000 / glid-3

  View on GitHub
  combination of OpenAI GLIDE and Latent Diffusion
  ☆136Apr 7, 2022Updated 4 years ago
• Sea-Snell / JAXSeq

  View on GitHub
  Train very large language models in Jax.
  ☆209Oct 21, 2023Updated 2 years ago
• hafriedlander / idea2art-aio

  View on GitHub
  Simple local all-in-one install for IDEA2.ART
  ☆26Jan 8, 2023Updated 3 years ago
• Wordpress hosting with auto-scaling - Free Trial Offer • Ad
  Fully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
• 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
• cgarciae / jax_metrics

  View on GitHub
  A metrics library for the JAX ecosystem
  ☆41Mar 16, 2023Updated 3 years ago
• ayaka14732 / bart-base-jax

  View on GitHub
  JAX implementation of the bart-base model
  ☆34Apr 11, 2023Updated 3 years ago
• instadeepai / fastpbrl

  View on GitHub
  Vectorization techniques for fast population-based training.
  ☆57Apr 26, 2026Updated last week
• Jack000 / guided-diffusion

  View on GitHub
  ☆28Dec 16, 2021Updated 4 years ago
• brentyi / jax_dataclasses

  View on GitHub
  Pytrees + dataclasses ❤️
  ☆75Dec 19, 2025Updated 4 months ago