microsoft / tutel

Tutel MoE: An Optimized Mixture-of-Experts Implementation
MIT License
694 stars 84 forks source link

add primitives: net.batch_all_to_all_v(), net.batch_all_gather_v() #221

Closed ghostplant closed 7 months ago