AI芯片量化工具研发工程师-校招(北京、合肥、无锡)
职位描述
1. 负责量化工具的核心功能开发与维护,包括但不限于自动化量化流程、模型解析与图优化、量化感知训练(QAT)及训练后量化(PTQ)等内容
2. 负责前沿模型量化算法的研究与实现,包括但不限于混合精度量化、非均匀量化、低比特(如4-bit/2-bit)量化等,并将其整合到自研工具链中
3. 针对不同的神经网络架构(如 CNN 和 Transformer)及其应用模型(如 LLM),面向特定硬件(自研芯片NPU)设计量化方案,优化量化策略,平衡模型精度、性能,解决量化过程中的精度损失问题
4. 负责提升工具的易用性、稳定性和可扩展性,构建完善的CI/CD流程,并撰写高质量的技术文档、教程和示例,为业务团队提供技术支持任职要求
1. 有良好的代码开发风格和软件工程思维,熟练使用Python
2. 熟悉主流神经网络算法原理,包括CNN、Transformer等
3. 熟悉常用量化算法,以及量化后的精度评估方法
满足以下条件者优先:
1. 熟悉PPQ、LLMC等开源量化框架
2. 熟悉Pytorch、TensorFlow等深度学习框架
3. 参与过AI开源项目或者编程竞赛,对领域内的知识有过思考、总结和分享等