机器学习系统研发工程师-Ads Infra
Core
Optimize computation graph execution efficiency for ad/recommendation/search scenarios and build high-performance training/inference engines.
Role type
Senior IC machine-learning systems engineer (ads infra)
Builds
High-performance training and inference engines, operator libraries, and distributed ML systems
Domain
Internet advertising, recommendation systems, search, GPU/NPU computing
Deliverable
production ML models
Required skills
C/C++/Python, Linux, data structures and algorithms, system design, distributed system principles, at least one ML framework (PyTorch/TensorFlow/PaddlePaddle)
Preferred skills
CUDA or Triton, TensorRT/CUTLASS, TRT-LLM/vLLM, XLA/MLIR/TVM, PyTorch FSDP/DeepSpeed/Megatron
Technologies
GPU, NPU, CUDA, Triton, TensorRT, CUTLASS, XLA, MLIR, TVM, PyTorch, TensorFlow, PaddlePaddle, Linux
Responsibilities
Optimize model training/inference computation graphs; Develop high-performance operators and libraries; Research and introduce cutting-edge hardware and heterogeneous computing technologies; Design and maintain distributed ML systems.
