xla
[SPMD] Patch nn.Linear
#5491
Merged

[SPMD] Patch nn.Linear #5491

alanwaketan merged 6 commits into master from alanwaketan/patch_linear_2
alanwaketan
alanwaketan Initial implementation
01e3c95b
alanwaketan Support bias
51300963
alanwaketan Support 2D
cdc16876
alanwaketan Adds 1D test
319ee1df
alanwaketan alanwaketan requested a review from yeounoh yeounoh 2 years ago
alanwaketan alanwaketan requested a review from jonb377 jonb377 2 years ago
alanwaketan alanwaketan requested a review from JackCaoG JackCaoG 2 years ago
alanwaketan alanwaketan assigned alanwaketan alanwaketan 2 years ago
alanwaketan minor fixes
1d7ddfa9
alanwaketan Fix linters
340326a0
JackCaoG
alanwaketan
JackCaoG
JackCaoG commented on 2023-08-24
JackCaoG
JackCaoG commented on 2023-08-24
JackCaoG
JackCaoG commented on 2023-08-24
JackCaoG
JackCaoG approved these changes on 2023-08-24
alanwaketan
alanwaketan alanwaketan merged 4f378e69 into master 2 years ago
yeounoh
alanwaketan

Login to write a write a comment.

Login via GitHub

Assignees
Labels
Milestone