Pull requests: AI-Hypercomputer/maxtext

New pull request New

235 Open 2,126 Closed

Pull requests list

feat: add qwen3-30b-a3b conversion compatibility

#2547 opened Oct 25, 2025 by emergenz

Add muon optimizer

#2546 opened Oct 25, 2025 by shuningjin • Draft

4 tasks

Add script to run evaluation before and after SFT training

#2545 opened Oct 24, 2025 by SurbhiJainUSC • Draft

4 tasks done

Integrate Diagon SDK for managed profiler

#2544 opened Oct 24, 2025 by xibinliu

4 tasks done

Refactor: 3.1 phase of RESTRUCTURE.md, focussing on setup.sh and related Python dependency & Dockerfile files

#2541 opened Oct 24, 2025 by SamuelMarks

4 tasks done

[WIP] Support scale down when loading grain checkpoint

#2537 opened Oct 22, 2025 by aireenmei • Draft

4 tasks done

Add user guide to update MaxText dependencies with seed-env

#2536 opened Oct 22, 2025 by bvandermoon

4 tasks done

Add rampup batch size support in MaxText gemini-review

#2535 opened Oct 22, 2025 by NuojCheng

4 tasks done

Fix: Correct path to extra_deps_from_github.txt

#2533 opened Oct 22, 2025 by AidenYu1673

4 tasks done

[Do Not Merge] Run Tests against jax 0.8.0 image

#2532 opened Oct 21, 2025 by Rohan-Bierneni • Draft

4 tasks done

GRPO refactoring

#2530 opened Oct 21, 2025 by mydatascience

4 tasks done

Qwen3-Next Gated FullAttention Implementation

#2529 opened Oct 21, 2025 by Rohan-Bierneni

4 tasks done

test ironwood

#2524 opened Oct 20, 2025 by A9isha • Draft

4 tasks

feat: migrate deepseek batch split to nnx pull ready

#2522 opened Oct 20, 2025 by mesakhcienet

4 tasks done

Fix exception causes in data_loader.py

#2521 opened Oct 18, 2025 by cool-RR

add weight_sum_fp32 config

#2519 opened Oct 17, 2025 by suexu1025

4 tasks done

New landing page

#2517 opened Oct 17, 2025 by melissawm

4 tasks done

A trial of apply @fused to optimzier state offloading

#2513 opened Oct 16, 2025 by zhenying-liu • Draft

Introduce gather() and scatter_add() reference functions for token-routing in Mixture-of-Experts models.

#2512 opened Oct 16, 2025 by copybara-service bot

feat(data): Default grain_worker_count to -1 for automatic performance tuning

#2510 opened Oct 16, 2025 by bzantium

4 tasks done

@aireenmei

Update patch to ensure maxtext images are downgraded to 0.7.0

#2506 opened Oct 15, 2025 by Rohan-Bierneni

4 tasks done

@Rohan-Bierneni

Fix JAX version bugs for nightly image

#2503 opened Oct 15, 2025 by hengtaoguo • Draft

4 tasks done

@hengtaoguo @gobbleturk

Fix generate_metrics_and_upload_to_big_query

#2502 opened Oct 15, 2025 by ycchenzheng

4 tasks done

@ycchenzheng

Recipe Modularization, DAG Integration, and BigQuery Integration

#2499 opened Oct 15, 2025 by DannyLiCom

4 tasks done

[*] pre-commit run --all-files

#2498 opened Oct 14, 2025 by SamuelMarks

4 tasks done

ProTip! Follow long discussions with comments:>50.

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Pull requests: AI-Hypercomputer/maxtext

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Pull requests list