β91Sep 19, 2022Updated 4 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.
Sorting:
- Train vision models using JAX and π€ transformersβ104Dec 14, 2025Updated 10 months ago
- JAX implementation of VQGANβ91Jul 9, 2022Updated 4 years ago
- JAX implementation ViT-VQGANβ82Sep 21, 2022Updated 4 years ago
- Unified API to facilitate usage of pre-trained "perceptor" models, a la CLIPβ38Nov 26, 2022Updated 3 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flaxβ162Nov 1, 2022Updated 4 years ago
- Deploy on Railway without the complexity - Free Credits Offer β’ AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Dreamer on JAXβ16Jan 19, 2022Updated 4 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objaxβ130Jan 4, 2024Updated 2 years ago
- Refactoring dalle-pytorch and taming-transformers for TPU VMβ60Aug 30, 2021Updated 5 years ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.β267Sep 4, 2026Updated 2 months ago
- Contrastive Language-Audio Pretrainingβ15May 18, 2021Updated 5 years ago
- Contrastive Language-Image Pretrainingβ148Sep 6, 2022Updated 4 years ago
- Official repository for Fourier model that can generate periodic signalsβ10Mar 10, 2022Updated 4 years ago
- High-Resolution Image Synthesis with Latent Diffusion Modelsβ60Apr 17, 2022Updated 4 years ago
- β28Dec 16, 2021Updated 4 years ago
- GPU virtual machines on DigitalOcean Gradient AI β’ AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Unofficial JAX implementations of deep learning research papersβ162Jun 25, 2022Updated 4 years ago
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the π€ Hugging Face hub.β16Aug 1, 2022Updated 4 years ago
- A JAX implementation of stochastic addition.β13Aug 15, 2022Updated 4 years ago
- β19Aug 19, 2021Updated 5 years ago
- Maximal Update Parametrization (ΞΌP) with Flax & Optax.β16Dec 27, 2023Updated 2 years ago
- jax version of clip guided diffusion scriptsβ90Jan 11, 2024Updated 2 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Trainingβ50Jan 20, 2024Updated 2 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variationβ12May 24, 2021Updated 5 years ago
- Implementation of numerous Vision Transformers in Google's JAX and Flax.β22Aug 30, 2022Updated 4 years ago
- AI Agents on DigitalOcean Gradient AI Platform β’ AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- JAX implementation of the Llama 2 modelβ217Feb 2, 2024Updated 2 years ago
- Official Implementation of Paella https://arxiv.org/abs/2211.07292v2β748Oct 4, 2023Updated 3 years ago
- A functional training loops library for JAXβ87Feb 13, 2024Updated 2 years ago
- β13Mar 27, 2020Updated 6 years ago
- This contains the Flax model of min(DALLΒ·E) and code for converting it to PyTorchβ45Jul 21, 2022Updated 4 years ago
- Simple local all-in-one install for IDEA2.ARTβ26Jan 8, 2023Updated 3 years ago
- A simple library for scaling up JAX programsβ149Nov 4, 2025Updated last year
- Visualize, create, and operate on pytrees in the most intuitive way possible.β48Aug 18, 2026Updated 2 months ago
- β372Apr 12, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer β’ AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Train very large language models in Jax.β209Oct 21, 2023Updated 3 years ago
- Pytrees + dataclasses β€οΈβ76Dec 19, 2025Updated 10 months ago
- Local Attention - Flax module for Jaxβ22May 26, 2021Updated 5 years ago
- Implementation of Vision Transformers in Flaxβ18Oct 12, 2020Updated 6 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodesβ241May 12, 2023Updated 3 years ago
- combination of OpenAI GLIDE and Latent Diffusionβ135Apr 7, 2022Updated 4 years ago
- adaptation of huggingface's dreambooth training script to support depth2imgβ100Dec 23, 2022Updated 3 years ago