Skip to content

Pull requests: AI-Hypercomputer/maxtext

New pull request New
221 Open 3,984 Closed
221 Open 3,984 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Add chunked MoE support to MaxText.
#4481 opened Jul 15, 2026 by copybara-service Bot Loading…
1
Fix QLoRA NNX issues
#4480 opened Jul 15, 2026 by hsuan-lun-chiang Collaborator • Draft
4 tasks done
1
Non-SPMD DiLoCo
#4479 opened Jul 15, 2026 by Dr-Left Collaborator • Draft
4 tasks
Fix packed-sequence masking in cudnn_flash_jax and add cutlass_flash GPU backend
#4478 opened Jul 15, 2026 by bzantium Collaborator Loading…
4 tasks done
1
1
fix: restore two-layer defense to load checkpoint
#4477 opened Jul 15, 2026 by mesakhcienet Collaborator • Draft
4 tasks
3
[WIP] robot arm experiment
#4475 opened Jul 15, 2026 by hengtaoguo Collaborator • Draft
4 tasks
1
Support TE Collective GEMM for Attention Layer Projections
#4470 opened Jul 15, 2026 by dinodeep Collaborator Loading…
4 tasks done
1
Optimize sequence reordering on TPU by eliminating jnp.take gather gemini-review
#4469 opened Jul 15, 2026 by JHCuc3m Collaborator • Draft
4 tasks done
13
Refine GPT-OSS 20B RL news entry in README.md
#4468 opened Jul 14, 2026 by susanbao Collaborator Loading…
4 tasks done
1
Fix CI: Run notebooks only on notebook changes
#4466 opened Jul 14, 2026 by NuojCheng Collaborator • Draft
4 tasks done
1
Use attention ops from Tokamax
#4465 opened Jul 14, 2026 by copybara-service Bot Loading…
1
1
Fix NNX sequential layer naming and PyTree structure
#4464 opened Jul 14, 2026 by xibinliu Collaborator • Draft
4 tasks done
1
Refactor RL tutorials
#4463 opened Jul 14, 2026 by copybara-service Bot Loading…
@SurbhiJainUSC
feat: Automated Post-Conversion Checkpoint Shape and Mock Tensor Vali…
#4462 opened Jul 14, 2026 by olufiyin19 Collaborator Loading…
4 tasks done
4
Decode with padded video masks for Qwen3 models
#4461 opened Jul 14, 2026 by hengtaoguo Collaborator Loading…
4 tasks done
6
add ragged gather reduce v2 pull ready
#4460 opened Jul 14, 2026 by NuojCheng Collaborator • Draft
4 tasks
1
Keep fused MoE weights intact through FSDP gather
#4459 opened Jul 14, 2026 by abhinavgoel95 Contributor Loading…
4 tasks done
Onboard glm5.1
#4458 opened Jul 14, 2026 by notabee Collaborator Loading…
4 tasks done
1
fix(test): improve test_scan_layers_mismatch_tpu assertion
#4457 opened Jul 14, 2026 by mesakhcienet Collaborator • Draft
4 tasks done
1
Remove deprecated new_slice_event compatibility property from pathwaysutils.
#4456 opened Jul 14, 2026 by copybara-service Bot Loading…
@lukebaumann
Relax tolerances in DiLoCo, MoE, and Attention tests to fix TPU7x failures
#4455 opened Jul 14, 2026 by hsuan-lun-chiang Collaborator Loading…
4 tasks done
1
Eliminate duplicated softmax recomputation in indexer loss. gemini-review
#4454 opened Jul 14, 2026 by JHCuc3m Collaborator Loading…
4 tasks done
6
Add multi-directory checkpoint loader for multimodal alignment poc
#4452 opened Jul 13, 2026 by subawocit Collaborator Loading…
4 tasks done
1
[draft] gmm v2 quantization pull ready
#4451 opened Jul 13, 2026 by shuningjin Collaborator • Draft
4 tasks
Abhinav/elastic snapshot
#4449 opened Jul 13, 2026 by abhinavclemson Collaborator Loading…
4 tasks
ProTip! no:milestone will show everything without a milestone.