上海为旌科技有限公司
AI芯片工具链开发工程师(南京)
RESPONSIBILITIES
岗位职责
1.基于深度学习的端侧AI量化工具链研发,打造从浮点模型到定点模型的高效转换解决方案
2.负责分析量化前后模型的精度差异与分布变化,提出量化精度优化方案
3.负责开发和维护应用于AI的量化工具,以支持异构系统对AI的加速,包括CPU/DSP/NPU
4.理解硬件和软件算法的具体需求,和芯片、软件同事密切合作,从量化角度参与硬件和软件系统的设计
5.负责端侧大模型(LLM/VLM)的部署工作,包括模型导出、格式转换及量化压缩,支撑大模型在资源受限场景下的高效落地
6.负责客户量化问题的技术支持,包括但不限于精度掉点排查、校准策略调优、算子异常诊断,驱动工具易用性持续改进
7.负责撰写相关技术文档及专利输出
REQUIREMENTS
任职要求
1.计算机、电子、通信、数学等相关专业硕士及以上学历,优秀本科生可放宽要求
2.扎实的Python及C++编程能力及数据结构与算法,熟悉软件设计与调试
3.熟练掌握且实际应用机器学习框架:PyTorch/TensorFlow/ONNX
4.熟悉AI计算流图以及各个算子的量化特性与敏感性
5.具备TensorRT/TVM/PyTorch Quantization等量化工具开发经验者优先
6.具备端侧大模型部署经验者优先
7.良好的英文阅读能力,团队合作能力,沟通能力和问题解决与推动能力
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。
