Skip to content

Offload MoE collectives and ragged-sort index math to the TPU SparseCore - #5140

Draft
NuojCheng wants to merge 3 commits into
mainfrom
chengnuojin-moe-sparsecore-offload
Draft

Offload MoE collectives and ragged-sort index math to the TPU SparseCore#5140
NuojCheng wants to merge 3 commits into
mainfrom
chengnuojin-moe-sparsecore-offload

Commits

Commits on Sep 4, 2026