LLM training code for Databricks foundation models
Role in this project:
ML Engineer Contributions:41 reviews, 28 PRs, 16 pushes in 1 year 5 months
Contributions summary:Tianshi primarily contributed to the core machine learning components of the repository, focused on improving the performance and stability of the language models. Their contributions included refactoring attention mechanisms, specifically implementing and optimizing Grouped Query Attention (GQA) and addressing edge cases in Triton-based attention. Furthermore, the user fixed bugs related to model initialization and padding, which improved the model's robustness. These changes were made to improve the foundation model training and inference capabilities.
deep-learningllmneural-networksnlppytorch
Contributions:201 pushes, 27 branches in 1 year 5 months