Refactor execution of batch-split deepseek sparse layers in Decoder to use pure JAX.#3182
Open
copybara-service[bot] wants to merge 1 commit intomainfrom
Open
Refactor execution of batch-split deepseek sparse layers in Decoder to use pure JAX.#3182copybara-service[bot] wants to merge 1 commit intomainfrom
copybara-service[bot] wants to merge 1 commit intomainfrom
Commits
Commits on Feb 20, 2026
- authored andcommitted
