fairscale/nn/moe/top2gate.py · 29d81c43fa06b3f23f7128a08950e678c26709f8 · OpenDAS / fairscale · GitLab

Find file Blame History Permalink

[perf] nn.moe: replace einsum with faster equivalent code (#667) · 29d81c43
msbaines authored May 08, 2021
```
Co-authored-by: @myleott
```
29d81c43

top2gate.py 4.64 KB