Skip to content

Fix explicit-mesh loss sharding assert on the NNX path#4208

Open
ecnal-cienet wants to merge 1 commit into
mainfrom
fix/deepseek-batchsplit-context-reshard
Open

Fix explicit-mesh loss sharding assert on the NNX path#4208
ecnal-cienet wants to merge 1 commit into
mainfrom
fix/deepseek-batchsplit-context-reshard

Reshard loss under explicit mesh on the NNX path

cd08cff
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Jun 22, 2026 in 7s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

cd08cff Author: @ecnal-cienet <lan*****ng​@cienet.com>

(Only the first commit for a unique contributor is listed.)