平头哥平头哥-深度学习算法框架高级技术专家-北京

任职要求

学历：计算机科学、电子工程、人工智能或相关领域的硕士以上学位。

经验：
至少5年以上深度学习算法研究或框架开发相关工作经验，具有丰富的实际项目经验。
熟悉一种或多种深度学习框架（如 TensorFlow、PyTorch、Keras 等）的开发和应用。
具备低精度（fp8，int8，fp4 等）量化算法经验

技术能力：
扎实的算法基础和编程能力，熟练掌握 Python、C++ 等编程语言。
深入理解主流深度学习框架Pytorch、Tensorflow 的内部实现原理；
理解分布式训练框架Megatron-LM、DeepSpeed 的特性，对大模型训练的5D 并行策略有深入理解；
熟悉大模型推理框架VLLM、SGLang等，并能够根据芯片硬件资源做性能优化；
具有AI相关软件开源社区工作经验者优先；
具有强化学习领域相关经验者优先；

个人素质：
具备较强的分析和解决问题的能力，良好的团队沟通和协作能力。
具有创新思维和自主学习能力，能够在快速变化的技术环境中保持领先。

工作职责

职位概述：
作为平头哥数据中心产品线深度学习算法框架高级技术专家，将负责设计、开发和优化深度学习算法框架，提升算法模型在PTG AI芯片上的运行效率和性能。跨团队协作，确保高质量的代码和技术实现，同时为团队提供技术指导。

主要职责：
算法框架适配与开发：
对开源AI软件框架的内部实现有深入理解，能够快速将广大开源软件适配到 AI 芯片上，同时确保性能达到最优。同时根据公司AI芯片的差异化的功能，在框架层面实现软硬件结合的解决方案。

性能优化：
分析和优化深度学习模型的训练和推理性能，提升计算效率和资源利用率。
根据芯片硬件特点使用多种技术手段，包括分布式计算、模型压缩和量化，优化算法在实际应用中的表现。

跨团队协作：
与其他团队（如编译器、驱动、客户支持等）紧密合作，确保技术指标的达成，并实现符合用户需求和市场趋势的功能。
为团队成员提供技术指导和支持，推动团队技能的持续提升。

技术研究与创新：
跟踪深度学习和机器学习领域的最新研究动态，探索并实施新算法和技术。
参与编写技术文档和论文，促进技术的分享与传播。