Commits · 29d81c43fa06b3f23f7128a08950e678c26709f8 · OpenDAS / fairscale

08 May, 2021 1 commit
- [perf] nn.moe: replace einsum with faster equivalent code (#667) · 29d81c43
  msbaines authored May 08, 2021
```
Co-authored-by: @myleott
```
  29d81c43
07 May, 2021 1 commit
- [perf] nn.moe: workaround inefficiency in PyTorch's one_hot (#666) · 99b30a04
  msbaines authored May 07, 2021
```
Workaround for https://github.com/pytorch/pytorch/issues/55579

Co-authored-by: @shruti-bh, @myleott
```
  99b30a04
12 Nov, 2020 1 commit
- [refactor] moe: cleanup code to be more readable (#186) · 34df6069
  msbaines authored Nov 11, 2020
  
  34df6069
11 Nov, 2020 1 commit
- [refactor] moe: remove G dimension (#183) · 89176e34
  msbaines authored Nov 11, 2020
  
  89176e34
06 Oct, 2020 1 commit
- [refactor] moe: simplify logic removing top expert (#125) · 6e7ad798
  msbaines authored Oct 05, 2020
  
  6e7ad798
05 Oct, 2020 1 commit
- [fix] moe: fix Top2Gate to work on GPU (#124) · 662667d0
  msbaines authored Oct 05, 2020
  
  662667d0
02 Oct, 2020 1 commit
- [feat] moe: initial implementation of Top2Gating (#118) · 7815f6f3
  msbaines authored Oct 01, 2020
  
  7815f6f3