T2 AI unicorns MiniMax Internship Round 2
MiniMax AI Infra internship second round
Real MiniMax AI Infra internship second-round interview questions covering training optimization and high-performance computing
Training optimization High-performance computing
Project questions
- Share an internship project, focusing on the hardest technical problem you hit and the optimization results you got
Theory
- What state does each DeepSpeed ZeRO stage (Stage-1 / Stage-2 / Stage-3) shard? How do they progress in communication volume and memory savings?
- Which parallelism schemes have you used for large-model training? What are the design ideas and fit conditions for data, tensor, and pipeline parallelism?
- To train a ~70B model, how do you roughly estimate memory needed on one GPU?
- What is the core idea of LoRA? Why does low-rank factorization cut the number of updated parameters so much?
- In LoRA, why are down-projection A and up-projection B initialized differently? What is that for?
- Besides ZeRO, what other large-model training optimizations do you know?
Coding
- Implement sliding-window maximum