Central place for the engineering/scaling WG: documentation, SLURM scripts and logs, compute environment and data.
Role in this project:
DevOps Engineer & ML Engineer Contributions:37 reviews, 732 commits, 7 PRs in 1 year 4 months
Contributions summary:Stas's contributions primarily revolve around setting up and configuring the computational environment for large language model training and related benchmarking. They've implemented and tested bandwidth benchmarks, and they've integrated a pre-trained model from a different repository. The user has also contributed to setting up the continuous integration process through adding tools for hub syncing and email alerts related to job status on Jean Zay, as well as created tools to monitor the status of Slurm jobs.
slurmnlpmachine-learningtrainingmodels
Ongoing research training transformer language models at scale, including: BERT & GPT-2
Role in this project:
ML Engineer Contributions:291 reviews, 334 commits, 176 PRs in 1 year 4 months
Contributions summary:Stas primarily contributed to the development and refinement of training scripts and related functionalities for transformer language models. Their work includes the addition and modification of scripts for training on single nodes with DeepSpeed, demonstrating an understanding of distributed training. The user also addressed and reverted issues with training sample calculations, and incorporated a merging datasets capability and a kill switch, enhancing the project's robustness and flexibility.
bertgptlanguage-model