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