DeepSpeed
adding TiledFusedLogitsLoss
#7437
Merged

adding TiledFusedLogitsLoss #7437

stas00 merged 6 commits into master from stas/TiledFusedLogitsLoss
stas00
stas00 adding TiledFusedLogitsLoss
f5dc59fe
stas00 add mask + some fixes
638fd50f
sfc-gh-aqiao support more complex use cases for SwiftKV
b1e18680
stas00 add docs
6d6bd566
stas00 stas00 marked this pull request as ready for review 59 days ago
stas00 stas00 requested a review from tjruwase tjruwase 59 days ago
stas00 stas00 requested a review from tohtana tohtana 59 days ago
stas00 stas00 requested a review from loadams loadams 59 days ago
stas00 Merge branch 'master' into stas/TiledFusedLogitsLoss
e9584a06
stas00 redo asserts
c7839162
stas00 stas00 enabled auto-merge (squash) 59 days ago
stas00
sfc-gh-truwase
sfc-gh-truwase approved these changes on 2025-07-30
stas00 stas00 merged 3292e07a into master 58 days ago
stas00 stas00 deleted the stas/TiledFusedLogitsLoss branch 58 days ago
sfc-gh-aqiao
sfc-gh-aqiao commented on 2025-07-30

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone