Skip to content

Refactor execution of deepseek sparse layers in Decoder to use pure JAX.#3182

Open
copybara-service[bot] wants to merge 1 commit intomainfrom
test_871881885
Open

Refactor execution of deepseek sparse layers in Decoder to use pure JAX.#3182
copybara-service[bot] wants to merge 1 commit intomainfrom
test_871881885

Conversation

@copybara-service
Copy link
Contributor

Refactor execution of deepseek sparse layers in Decoder to use pure JAX.

@codecov
Copy link

codecov bot commented Feb 18, 2026

Codecov Report

❌ Patch coverage is 31.03448% with 20 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
src/MaxText/layers/deepseek_batchsplit.py 14.28% 12 Missing ⚠️
src/MaxText/layers/deepseek.py 25.00% 6 Missing ⚠️
src/MaxText/layers/decoders.py 50.00% 1 Missing and 1 partial ⚠️

📢 Thoughts on this report? Let us know!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

0 participants

Comments