Support nn.EmbeddingBag#519

Closed

alexandresablayrolles wants to merge 6 commits intomainfrom

pass_all_inputs

Commits on Oct 31, 2022

Pass activations as a list
Alex Sablayrolles
committed
Add embeddingbag grad sampler
Alex Sablayrolles
committed
Finish tests. Einsums work for embeddingbag. Functorch is disabled for embeddingbag as it is difficult to make it work.
Alex Sablayrolles
committed
[ahem] Committing embedding bag tests
Alex Sablayrolles
committed
Fix lint
Alex Sablayrolles
committed

Commits on Nov 2, 2022

Remove EW from test and fix lint
Alex Sablayrolles
committed