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

职位概述： 作为平头哥数据中心产品线深度学习算法框架高级技术专家，将负责设计、开发和优化深度学习算法框架，提升算法模型在PTG AI芯片上的运行效率和性能。跨团队协作，确保高质量的代码和技术实现，同时为团队提供技术指导。 主要职责： 算法框架适配与开发： 对开源AI软件框架的内部实现有深入理解，能够快速将广大开源软件适配到 AI 芯片上，同时确保性能达到最优。同时根据公司AI芯片的差异化的功能，在框架层面实现软硬件结合的解决方案。 性能优化： 分析和优化深度学习模型的训练和推理性能，提升计算效率和资源利用率。 根据芯片硬件特点使用多种技术手段，包括分布式计算、模型压缩和量化，优化算法在实际应用中的表现。 跨团队协作： 与其他团队（如编译器、驱动、客户支持等）紧密合作，确保技术指标的达成，并实现符合用户需求和市场趋势的功能。 为团队成员提供技术指导和支持，推动团队技能的持续提升。 技术研究与创新： 跟踪深度学习和机器学习领域的最新研究动态，探索并实施新算法和技术。 参与编写技术文档和论文，促进技术的分享与传播。 【任职要求】 学历：计算机科学、电子工程、人工智能或相关领域的硕士以上学位。 经验： 至少5年以上深度学习算法研究或框架开发相关工作经验，具有丰富的实际项目经验。 熟悉一种或多种深度学习框架（如 TensorFlow、PyTorch、Keras 等）的开发和应用。 具备低精度（fp8，int8，fp4 等）量化算法经验 技术能力： 扎实的算法基础和编程能力，熟练掌握 Python、C++ 等编程语言。 深入理解主流深度学习框架Pytorch、Tensorflow 的内部实现原理； 理解分布式训练框架Megatron-LM、DeepSpeed 的特性，对大模型训练的5D 并行策略有深入理解； 熟悉大模型推理框架VLLM、SGLang等，并能够根据芯片硬件资源做性能优化； 具有AI相关软件开源社区工作经验者优先； 具有强化学习领域相关经验者优先； 个人素质： 具备较强的分析和解决问题的能力，良好的团队沟通和协作能力。 具有创新思维和自主学习能力，能够在快速变化的技术环境中保持领先。