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 NumPy 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.
🚀 2-4x faster PyTorch training with one line of code. Beats torch.compile by 79%. Zero config, automatic hardware optimization for T4/V100/A100/H100 GPUs.
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.
Profile-first CUDA lab for killing GPU pipeline bubbles: stream overlap, CUDA Graphs, torch.compile, spec-decode overlap — Nsight-verified
Profiling CUDA Graph vs torch.compile vs eager decode on RTX 2070 — measuring per-token latency reduction across batch sizes, sequence lengths, and GPT-2 model sizes.
PyTorch 2.9 + HIP/ROCm learning notes: dispatcher, accelerator backend, profiler, TorchInductor, runtime, GEMM, and C++ extensions.
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.
PyTorch GPU inference performance exercises: roofline analysis, decode-loop profiling, KV-cache optimization, torch.compile, and CUDA graphs.
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.
Plain English + a PyTorch reference → a verified, benchmarked CUDA kernel you can pip install and torch.compile.
One instrumentation engine for PyTorch training, two products: a fault injection tested health monitor (NaN, gradient explosion, stall, with diagnosis and ranked remedies) and a throughput sweeper (batch size, precision, DataLoader, torch.compile) with auditable bottleneck verdicts. 1.90% overhead.
A torch.compile backend that pays attention: custom FX fusion passes, a Triton LayerNorm kernel, and a persistent compile cache. Apache-2.0 free core of the g2n platform.
Add a description, image, and links to the torch-compile topic page so that developers can more easily learn about it.
To associate your repository with the torch-compile topic, visit your repo's landing page and select "manage topics."