A new library called KOSMOS has been developed to improve the efficiency of training large language models on GPUs. KOSMOS optimizes the communication and computation phases within transformer layers by using self-scheduling persistent kernels that adapt to data readiness. When evaluated on AMD Instinct MI350X and MI355X GPUs, KOSMOS demonstrated significant speedups, ranging from 1.10x to 1.83x, for models like Megatron-LM, DeepSeek-V3, and Qwen3-235B compared to existing methods. AI
IMPACT This library could significantly reduce LLM training times and costs, making large model development more accessible.
RANK_REASON The cluster describes a new library and its performance evaluation presented in an arXiv paper. [lever_c_demoted from research: ic=1 ai=1.0]
- AMD Instinct MI350X
- AMD Instinct MI355X
- arXiv
- DeepSeek-V3
- Hugging Face
- KOSMOS
- Megatron-LM
- Qwen3 235B
AI-generated summary · Google Gemini · from 1 sources. How we write summaries →