xla
Fully revamp aten embedding bag for JAX backend
#8535

Merged

Fully revamp aten embedding bag for JAX backend #8535

qihqi merged 1 commit into pytorch:master from vyom1611:test-embedding-aten

Refactor aten-embedding bag according to TODOs

c83b7974

qihqi dismissed these changes on 2025-01-06

qihqi dismissed their stale review 300 days ago

i'll run ci first

qihqi requested a review from

qihqi 300 days ago

qihqi approved these changes on 2025-01-07

qihqi merged 5278e7a5 into master 299 days ago

Reviewers

qihqi

Assignees

No one assigned

Labels

None yet

Milestone

No milestone