kwai / Megatron-Kwai

[USENIX ATC '24] Accelerating the Training of Large Language Models using Efficient Activation Rematerialization and Optimal Hybrid Parallelism
53Updated 9 months ago

Alternatives and similar repositories for Megatron-Kwai:

Users that are interested in Megatron-Kwai are comparing it to the libraries listed below