ROCm TransformerEngine wheels 2026-06-22 (TE 9c19ec753b2d)

Built from ROCm/TransformerEngine dev @ 9c19ec753b2d on 2026-06-22.

ROCm: 7.2.0
Python: cp312
Arch: mi355 (gfx=gfx950)

ROCm TransformerEngine wheels (latest)

Rolling release for latest weekly-built ROCm TransformerEngine wheels used by CI.

rocm-maxtext-v0.1.1

This is the first downstream release of MaxText intended to mirror upstream MaxText while enabling ROCm-focused CI. It includes decoupled mode support (no Google Cloud required) and introduces ROCm CI jobs to validate the downstream stack on AMD GPUs.

What’s included

ROCm CI jobs (MI300) added to the downstream workflow to run MaxText unit tests on ROCm.
Decoupled-mode testing to ensure MaxText can run without Google Cloud services/dependencies.
Decoupled requirements files for JAX 0.8.2 (CPU + ROCm variants), aligned to upstream dependency style while keeping decoupled behavior.
Workflow improvements to support ROCm containers/runners and decoupled execution in CI.

Why this release exists

Downstream goal: establish reliable CI coverage for ROCm and decoupled mode while mirroring upstream behavior.
Upstream goal: once the downstream ROCm CI is stable, upstream equivalent ROCm CI changes will be proposed.

Assets

This release includes:
MaxText wheel (maxtext-*.whl)
Transformer Engine wheel (published alongside for CI convenience)

Notes

This release is focused on CI enablement and infrastructure alignment, not model/algorithm feature changes.
Decoupled mode is a shared direction for both upstream and downstream; this release validates it in the downstream ROCm environment.