An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries
Role in this project:
ML Engineer Contributions:1 review, 6 commits, 2 PRs in 1 year 2 months
Contributions summary:Vincent contributed to the implementation and improvement of a model parallel autoregressive transformer, specifically within the context of GPT-NeoX. Their work included integrating Flash Attention, a technique for optimizing attention mechanisms. They also made adjustments to Alibi matrix caching for dynamic sequence lengths and refined other attention-related configurations. Further contributions include the addition of maximum tokens to the interactive mode.
deepspeeddeepspeed-librarygpt-3transformerslanguage-model
Contributions:2 commits, 3 PRs, 3 pushes in 1 year 3 months