# MaxDiffusion ## Docs - [Checkpointing](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/advanced/checkpointing.md): Save and restore model states with Orbax checkpointer - [LoRA inference](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/advanced/lora.md): Load and use LoRA adapters with Flux and Wan models - [Multi-LoRA loading](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/advanced/multi-lora.md): Load and combine multiple LoRA adapters for enhanced customization - [Profiling](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/advanced/profiling.md): Profile training and inference performance with JAX profiler - [Quantization](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/advanced/quantization.md): Reduce model memory footprint with quantization - [Base configuration](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/config/base-config.md): Core configuration parameters for MaxDiffusion models - [Parallelism configuration](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/config/parallelism.md): Configuration for distributed training and inference with JAX - [Scheduler configuration](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/config/scheduler.md): Configuration for noise schedulers in MaxDiffusion - [generate.py](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/generate.md): Stable Diffusion 2.x inference script for generating images - [generate_flux.py](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/generate-flux.md): Flux inference script for state-of-the-art image generation - [generate_ltx_video.py](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/generate-ltx-video.md): LTX-Video generation script for text-to-video and image-to-video synthesis - [generate_sdxl.py](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/generate-sdxl.md): Stable Diffusion XL inference script for high-resolution image generation - [generate_wan.py](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/generate-wan.md): Wan video generation script for text-to-video and image-to-video synthesis - [ControlNet model](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/models/controlnet.md): Reference documentation for the FlaxControlNetModel architecture - [Transformer models](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/models/transformer.md): Reference documentation for transformer architectures in MaxDiffusion - [UNet 2D condition model](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/models/unet.md): Reference documentation for the FlaxUNet2DConditionModel architecture - [Variational autoencoder (VAE)](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/models/vae.md): Reference documentation for VAE models in MaxDiffusion - [Flux pipeline](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/pipelines/flux.md): Reference documentation for the Flux pipeline - [Stable Diffusion XL pipeline](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/pipelines/sdxl.md): Reference documentation for the SDXL pipeline - [Stable Diffusion pipeline](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/pipelines/stable-diffusion.md): Reference documentation for the Stable Diffusion pipeline - [WAN pipeline](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/pipelines/wan.md): Reference documentation for the WAN video generation pipeline - [train.py](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/train.md): Train Stable Diffusion 2.x models on TPU and GPU - [train_flux.py](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/train-flux.md): Train Flux Dev models on TPU v5p - [train_sdxl.py](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/train-sdxl.md): Train Stable Diffusion XL models on TPU and GPU - [train_wan.py](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api/train-wan.md): Train Wan 2.1 and 2.2 text-to-video models on TPU - [Architecture](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/concepts/architecture.md): Understanding MaxDiffusion's architecture and design principles - [Attention mechanisms](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/concepts/attention-mechanisms.md): Understanding attention implementations in MaxDiffusion - [Models](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/concepts/models.md): Supported diffusion models in MaxDiffusion - [Parallelism](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/concepts/parallelism.md): Understanding parallelism strategies in MaxDiffusion - [DGX Spark deployment](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/deployment/dgx-spark.md): Run MaxDiffusion on NVIDIA DGX Spark GPU workstations - [Multihost deployment](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/deployment/multihost.md): Scale MaxDiffusion training across multiple TPU hosts - [Single host deployment](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/deployment/single-host.md): Run MaxDiffusion on a single TPU host for local development - [XPK deployment](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/deployment/xpk.md): Deploy MaxDiffusion on GKE using XPK for production workloads - [Data preprocessing](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/guides/data-preprocessing.md): Prepare datasets for training MaxDiffusion models - [Optimization](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/guides/optimization.md): Performance optimization techniques for MaxDiffusion - [Troubleshooting](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/guides/troubleshooting.md): Common issues and solutions for MaxDiffusion - [ControlNet inference](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/inference/controlnet.md): Conditional image generation with edge detection and other control signals - [Flux inference](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/inference/flux.md): Fast, high-quality image generation with Flux dev and schnell variants - [LTX Video generation](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/inference/ltx-video.md): Generate high-quality videos with LTX-Video text-to-video and image-to-video models - [Inference overview](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/inference/overview.md): Generate images and videos with MaxDiffusion's optimized inference pipelines - [Stable Diffusion XL inference](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/inference/sdxl.md): Generate high-quality 1024×1024 images with SDXL and variants - [Stable Diffusion inference](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/inference/stable-diffusion.md): Generate 512×512 images with Stable Diffusion 2 base and 2.1 - [Wan video generation](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/inference/wan.md): Generate high-quality videos with Wan 2.1 and 2.2 text-to-video and image-to-video models - [Installation](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/installation.md): Set up MaxDiffusion on Cloud TPUs, GPUs, or local development - [Introduction](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/introduction.md): High-performance latent diffusion models in pure Python/JAX for training and inference on XLA devices - [Quickstart](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/quickstart.md): Generate your first image with MaxDiffusion in minutes - [Dreambooth training](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/training/dreambooth.md): Personalized fine-tuning for Stable Diffusion models - [Flux training](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/training/flux.md): Fine-tune Flux transformer-based diffusion models - [Training overview](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/training/overview.md): Learn how to train and fine-tune diffusion models with MaxDiffusion - [Stable Diffusion XL training](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/training/sdxl.md): Fine-tune Stable Diffusion XL models at 1024x1024 resolution - [Stable Diffusion training](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/training/stable-diffusion.md): Train Stable Diffusion 1.4 and 2 Base models on TPUs and GPUs - [Wan 2.1 training](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/training/wan.md): Train Wan 2.1 video generation models for text-to-video and image-to-video ## OpenAPI Specs - [openapi](https://mintlify.wiki/AI-Hypercomputer/maxdiffusion/api-reference/openapi.json)