强化学习训练框架工程师（火山引擎机器学习平台）-Data AML
- 北京市
- 长期
- 全职
2、veOmini LLM/DiT训练框架研发与优化：负责veOmini中LLM与DiT训练模块的开发与优化，支持Wan2.1/Qwen/DeepSeek等热点模型的高效后训练和强化学习。职位要求:1、熟练掌握Linux环境下C/C++、Python编程，具备高效的代码实现与调试能力；
2、深度掌握PyTorch框架，掌握Profiling、调优方式，了解其底层原理；
3、具备训练框架（如Megatron-LM、DeepSpeed、TorchTuner）或推理框架（如vLLM、SG-Lang、TensorRT）的开发和优化经验；