Stefan He 8ecf6b9d24 Support Flatten Tensor Update Weights to speed up MOE Update Weights by 20% (#8079)

2025-08-10 16:08:59 -07:00

..

tensor_bucket.py

Support Flatten Tensor Update Weights to speed up MOE Update Weights by 20% (#8079)

2025-08-10 16:08:59 -07:00

utils.py

Add tensor.detach() back to update weight util (#8691)

2025-08-02 00:41:05 -07:00