机器学习研发工程师-搜索架构
Posted Updated 6
views
Description
1、负责搜索架构系统的开发,支撑团队相关业务的算法生产与高效迭代;
2、设计和实现机器学习相关的基础设施/算法框架/工具链等,维护稳健的调用链路,并推动落地到业务中;
3、支持大规模任务调度与分布式计算环境的构建与维护;
4、设计和开发高性能算子库,研究最新的GPU计算技术和优化方法,充分利用GPU并行计算能力,支持各种模型在不同硬件上的构建和部署。
1、掌握扎实的计算机基础知识,深入理解数据结构、算法和操作系统知识;
2、有强烈的求知欲,快速学习和沟通能力;
3、编程能力扎实,有良好的编程习惯。
加分项:
1、熟悉主流的深度学习框架(PyTorch、TensorFlow);
2、熟悉主流高性能并行编程技术,有GPU编程经验(CUDA或Triton);
3、熟悉vLLM,TRT-LLM等大模型推理框架,熟悉大模型并发推理原理;
4、熟悉FSDP、DeepSpeed、Megatron等业界主流分布框架的使用和原理。