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 makes it fast! ⚡
Wide-model collective ensemble system with fractal, geometric, and heavy compilation optimizations.
Leveraging torch.compile to accelerate cross-encoder inference
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
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.
🚀 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.
Profile-first CUDA lab for killing GPU pipeline bubbles: stream overlap, CUDA Graphs, torch.compile, spec-decode overlap — Nsight-verified
PyTorch GPU inference performance exercises: roofline analysis, decode-loop profiling, KV-cache optimization, torch.compile, and CUDA graphs.
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.
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.
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."