xla
[Pallas] Introduce GMM(torch.autograd.Function)
#7152

Merged

[Pallas] Introduce GMM(torch.autograd.Function) #7152

alanwaketan merged 3 commits into master from alanwaketan/tgmm4

Introduce GMM

4baa7361

Use torch.autograd.backward

4dd8826f

Fix linters

f7131d98

alanwaketan added tpuci

alanwaketan requested a review from

miladm 1 year ago

alanwaketan requested a review from

wonjoo-wj 1 year ago

alanwaketan requested a review from

JackCaoG 1 year ago

alanwaketan assigned

alanwaketan 1 year ago

JackCaoG commented on 2024-05-30

JackCaoG approved these changes on 2024-05-30

alanwaketan merged aeed61a9 into master 1 year ago

alanwaketan deleted the alanwaketan/tgmm4 branch 1 year ago

Reviewers

JackCaoG

miladm

wonjoo-wj

Assignees

alanwaketan

Labels

None yet

Milestone

No milestone