岗位描述
岗位职责
职位描述:
我们正在寻找一名专注于模型推理与模型优化的机器学习工程师。你将从模型结构和算法层面出发,对大规模深度学习模型进行优化,使其能够高效部署在小鹏定制的 AI 加速器上。
我们通过数据驱动的模型压缩与结构改造,在保证精度的前提下显著降低推理延迟和内存占用。
从模型和算法层面对深度学习模型进行推理优化,支持其在小鹏定制 AI 加速器上的高效部署
我们的使命是解决自动驾驶,具身职能这一世界级难题。你将与机器学习工程师、编译器工程师和研究科学家紧密合作,将前沿的大模型研究成果转化为可落地的工程方案。
岗位要求
职位要求:
CS/CE/EE硕士学位,或同等学历,行业经验
精通 Python 和 PyTorch,具备良好的研究型和工程型代码能力。
对大模型结构(如 Transformer 及其变体)有深入理解。
对模型推理框架和工具(onnx,TensorRT, TRT-LLM等)有深入了解。
扎实的深度学习基础和推理流程认知。
具备模型优化经验,如量化、剪枝、稀疏化、蒸馏等。
具备较强的自主学习和论文阅读能力,能够独立推动问题解决。
对挑战性和不确定性问题有好奇心,责任感和执行力。