Transformer Engine prebuilt Wheel ## Official Release, prebuilt wheel ``` 260121_a/transformer_engine-2.5.0+f05f12c9-cp312-cp312-linux_x86_64.whl * build environment: nvcr.io/nvidia/pytorch:25.06-py3 * torch: 2.8.0a0+5228986c39.nv25.06 * cuda : 12.9 * cudnn: 91002 (9.10.2) * TE branch release_v2.11 c188b5 dec 16 (CUDA_ARCHS="90") ``` ``` 250823_a/transformer_engine-2.5.0+f05f12c9-cp312-cp312-linux_x86_64.whl * build environment: nvcr.io/nvidia/pytorch:25.06-py3 * torch: 2.8.0a0+5228986c39.nv25.06 * cuda : 12.9 * cudnn: 91002 (9.10.2) * TE stable v2.5 (CUDA_ARCHS="90;100;120") ``` ## Unofficial, prebuilt wheel for local experimentation ``` unofficial/251115_a/transformer_engine-2.5.0+c2056e01-cp312-cp312-linux_x86_64.whl * build environment: nvcr.io/nvidia/nemo:25.07 torch: 2.8.0a0+5228986c39.nv25.05 cuda : 12.9 cudnn: 91100 * TE vuiseng9's fork:v2.5+nv4f_mx8b (CUDA_ARCHS=100) ``` ``` unofficial/251025_a/transformer_engine-2.8.0+7cde9a33-cp312-cp312-linux_x86_64.whl * build environment: nvcr.io/nvidia/nemo:25.09 torch: 2.8.0a0+5228986c39.nv25.06 cuda : 12.9 cudnn: 91100 (9.11.0) * TE vuiseng9's fork:v2.8 ++ (CUDA_ARCHS=100a) ``` ``` unofficial/251020_a/transformer_engine-2.5.0+c2056e01-cp312-cp312-linux_x86_64.whl * build environment: nvcr.io/nvidia/nemo:25.07 * torch: 2.8.0a0+5228986c39.nv25.05 * cuda : 12.9 * cudnn: 91100 (9.11.0) * TE: vuiseng9's fork: branch v2.5+nvfp4 (c2056e01) # CUDA_ARCHS="100" ``` ``` unofficial/251015_a/transformer_engine-2.5.0+010e0427-cp312-cp312-linux_x86_64.whl * build environment: nvcr.io/nvidia/pytorch:25.06-py3 * torch: 2.8.0a0+5228986c39.nv25.06 * cuda : 12.9 * cudnn: 91002 (9.10.2) * TE: vuiseng9's fork: branch v2.5+nvfp4 (010e0427) # CUDA_ARCHS="100" ```