Motivation

Follows up on #20170 review feedback. Moves platform-specific AITer GroupNorm
replacement out of decoding.py into the Platform polymorphic pattern:

• Register SGLANG_USE_ROCM_VAE in envs.py
• Add optimize_vae() hook to Platform base class
• Override in RocmPlatform with AITer GroupNorm replacement
• Call from VAELoader at load time
• Remove all platform-specific logic from decoding.py

Modifications

Accuracy Tests

Benchmarking and Profiling

Checklist

• Format your code according to the Format code with pre-commit.
• Add unit tests according to the Run and add unit tests.
• Update documentation according to Write documentations.
• Provide accuracy and speed benchmark results according to Test the accuracy and Benchmark the speed.
• Follow the SGLang code style guidance.

Review Process

1. Ping Merge Oncalls to start the PR flow. See the PR Merge Process.
2. Get approvals from CODEOWNERS and other reviewers.
3. Trigger CI tests with comments or contact authorized users to do so.
   • /tag-run-ci-label, /rerun-failed-ci, /tag-and-rerun-ci
4. After green CI and required approvals, ask Merge Oncalls to merge.