Gary Miguel 07887f6630 sharded_moe: make top1gating ONNX-exportable (#1578) | 2 years ago | |
---|---|---|
.. | ||
__init__.py | f28432441b DeepSpeed MoE (#1310) | 3 years ago |
experts.py | f28432441b DeepSpeed MoE (#1310) | 3 years ago |
layer.py | 2887349cd4 Adding Tutel to MoE layer (#1528) | 3 years ago |
sharded_moe.py | 07887f6630 sharded_moe: make top1gating ONNX-exportable (#1578) | 2 years ago |
utils.py | fc2f378ece Improve pre-commit hooks (#1602) | 2 years ago |