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