Machine Learning Engineer Intern

岗位职责
我们希望你具备：
- 具备机器学习或高性能图形处理经验。
- 对至少一种机器学习框架（如 PyTorch、JAX）有扎实的实践理解。
- 具备用 Python 和/或 C++ 编写优雅且易于维护的代码的强大能力。
- 能够快速学习并深入理解新概念或复杂的代码库。
- 具备以性能和效率为导向的思维方式，并对最细微的细节抱有浓厚兴趣。
- 具备在全球分布式团队中工作的强大沟通能力。

岗位要求
职位要求
进阶期待
- 对探索 PyTorch 内部机制充满热情，并在 torch.compile、fully_shard (FSDP2) 等 API 领域具备实践经验。
- 具备构建 Triton 内核的经验。
- 具备大规模分布式训练经验，熟悉现代并行化技术：DP、TP、CP、PP、零冗余优化器等。
- 具备3D或视频扩散模型相关经验
- 具备低精度bf16或fp8训练经验