End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 training).
-
Updated
Jan 8, 2026 - Python
End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 training).
Profine automatically profiles and optimizes PyTorch training jobs on real GPUs, delivering measurable speedups and lower GPU costs before teams waste days tuning configs by hand.
PyTorch extension for Rebellions NPU
How does torch.compile() achieve massive speedups despite highly optimized torch functions? See how operator fusion works!
Wide-model collective ensemble system with fractal, geometric, and heavy compilation optimizations.
Optimized CSM-1B TTS pipeline for RTX 5090 (Blackwell sm_120). CUDA graph replay via patched HF Transformers. ~0.46x RTF. Topics (tags): csm text-to-speech rtx-5090 blackwell cuda-graphs torch-compile sesame streaming pytorch
Leveraging torch.compile to accelerate cross-encoder inference
Compression-aware Qwen3-VL inference engine with torch.compile, CUDA Graph, scaled-FP8 KV, visual compaction, and multimodal prefix caching.
Benchmarking, optimising and serving the TabFM tabular foundation model online: exact KV cache, torch.compile, FastAPI, Triton, KServe on kind and GKE
Distilling Florence-2 into a compact ~60M-param VQA student (feature-matching + attention transfer) — ~70% teacher-accuracy retention at 138x inference throughput with bf16 + torch.compile on H100.
Evidence-gated torch.compile optimization lab with offline-verifiable bundles
Measurement-driven PyTorch training autotuner — tries precision × torch.compile × memory-format configs on your model, keeps the fastest, never returns one slower than your baseline. Honest, committed benchmarks.
Reproducible benchmark harness for neural TTS inference. Isolated Docker A/B arms, compiler vs eager measurement, request-path profiling, cross-process determinism controls, VRAM soak testing, and an ASR-scored regression suite that catches reference-audio leakage. Measured on OpenAudio S1-mini, RTX 5080.
Optimized PyTorch inference for NISQA speech quality and TTS naturalness models
One-command trust auditor for torch.compile: eager vs aot_eager vs inductor, five correctness checks, minimal repro, paste-ready report.
GPU systems benchmark comparing PyTorch eager execution and torch.compile across common deep learning operations. Measures execution time, throughput, TFLOPS, memory bandwidth, and kernel-level speedups through a reproducible benchmarking pipeline and publication-ready visualizations.
Guards a real torch.compile(inductor) CPU precision bug: std/var_mean/std_mean accumulate in float32 instead of eager's double precision (pytorch/pytorch#197089)
Three measured notebooks on GPU inference optimization: roofline analysis, KV-cache decode optimization (4.21x), and CUDA-graph launch-overhead elimination (5.38x). Pure PyTorch.
PyTorch 2.9 + HIP/ROCm learning notes: dispatcher, accelerator backend, profiler, TorchInductor, runtime, GEMM, and C++ extensions.
To associate your repository with the torch-compile topic, visit your repo's landing page and select "manage topics."