SA3-medium DiT — FP8 TensorRT artifacts

Contents

• onnx/sa3-m/dit_fp8.onnx + dit_fp8.onnx.data — the quantized ONNX (arch-independent; compile with build_from_onnx.py sa3-m-fp8, plain STRONGLY_TYPED, no ModelOpt needed)
• tensorRT/sm_120/sa3-m/dit_fp8.trt — prebuilt engine for RTX 50xx (sm_120), TensorRT 10.16.1.11. TRT engines are not portable across GPU architectures or TRT minor versions; rebuild from the ONNX for anything else.

Validation (vs the FP16-mixed engine, 47 prompts x 8 sigmas, L=646, RTX 5090)

• worst single-step latent cosine (x + dt*v, n=376): 0.9982
• 8-step compounded euler final-latent cosine over 47 prompts: mean 0.953, median 0.957, worst 0.873 (the rollout is chaotic; a 1e-3 input perturbation alone compounds to ~0.967, so this is a guide, not a gate)
• decoded audio under the production pingpong sampler tracks the FP16-mixed generation at ~0.90 RMS-curve correlation (same conditioning and seeds) and was validated by listening: the published sm_120 engine here is the exact engine that passed that test
• step latency B=1 L=646: ~10.6-11.0 ms vs ~18.7-19.4 ms FP16-mixed (~1.8x)
• under the stochastic pingpong sampler the engine produces a different but comparable sample