机器学习性能工程师-北京(A97980)
20000-40000/天
机器学习性能工程师-北京(A97980) 20000-40000/天
北京
硕士
0
发布于 6月30日
职位描述
PythonTensorFlowPyTorch硬件
职责描述: 1. 理解复杂的业务挑战,参与语言、视觉及多模态算法模型的训练和推理优化; 2. 研究当前业界领先的模型训练和推理加速方法(5D Parallelism,Attention 加速,KV Cache,Speculation,etc…)和模型轻量化(MoE,压缩,量化,etc…),不断提升算法性能; 3. 将研究从概念转化为部署和实施,开展算法和算力分析,促进软硬件协同优化工作。 任职要求: 1. 计算机科学、机器学习、数据科学或相关领域的硕士及以上学历; 2. 熟悉常用的数据结构和算法模型,熟悉 LLM/VLM/VLA 的训练和微调方法; 3. 熟练使用 PyTorch,JAX,TensorFlow 等框架; 4. 精通 Python/C/C++/CUDA 编程,具备数据结构和算法方面的实际经验; 5. 有大规模分布式并行优化,AI 算子优化,AI 编译优化等方面的经验。
工作地址
北京市/北京市/顺义区 理想汽车北京研发总部C区
相似职位推荐