Skip to content

Pull requests: InfiniTensor/InfiniTrain

New pull request New
15 Open 115 Closed
15 Open 115 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

【训练营】2026.3.16 flashattention
#131 opened Mar 17, 2026 by xuetongchenbianchen Loading…
@kilinchange
1
【训练营】End
#130 opened Mar 17, 2026 by xuetongchenbianchen Loading…
@kilinchange
1
【训练营】Checkpoint 读取工具
#129 opened Mar 17, 2026 by ArcaLunar Loading…
@JYMiracle305
4
【训练营】FlashAttention接入 @simon_chou
#128 opened Mar 16, 2026 by Simon-CHOU Loading…
@kilinchange
1
【训练营】BF16 多算子 (Linear, Fill, Cast, AccumulateGrad) Kernel 优化
#127 opened Mar 16, 2026 by xxycasc Loading…
@Ziminli
【训练营】BF16 Elementwise Kernel 优化
#126 opened Mar 16, 2026 by WjhisTy Loading…
@Ziminli
【训练营】FlashAttention 接入
#125 opened Mar 16, 2026 by BAI-123-GUO Loading…
@kilinchange
【训练营】Add FlashAttention operator into infiniTrain Framework
#124 opened Mar 16, 2026 by Aoshine999 Loading…
@kilinchange
1
【训练营】FlashAttention 接入与测试报告
#123 opened Mar 16, 2026 by jiewcc Loading…
@kilinchange
2
【训练营】基于 CTest + gtest 的测试体系搭建与工程化集成
#120 opened Mar 16, 2026 by luoyueyuguang Loading…
@chen2021673
【训练营】feat: integrate Flash Attention operator into InfiniTrain framework
#119 opened Mar 16, 2026 by tangguochuan Loading…
@kilinchange
2
feat: extract the common module of Transformer
#115 opened Mar 13, 2026 by JYMiracle305 Loading…
1
[WIP] dtype registry
#114 opened Mar 12, 2026 by kilinchange Loading…
【训练营】学习率调度器实现
#113 opened Mar 11, 2026 by littleotherut Loading…
@Chamberlain0w0
9
[WIP] feat: Support ZeRO-2 based on DistributedOptimizer
#110 opened Feb 28, 2026 by Chamberlain0w0 Loading…
ProTip! Updated in the last three days: updated:>2026-03-14.