Motional Singapore is seeking a Senior ML Systems Engineer to enhance the performance and optimization of machine learning systems. The ideal candidate will have a strong background in software engineering, particularly with Python and PyTorch, and experience in optimizing machine learning workflows
Job Summary
This role focuses on identifying and resolving bottlenecks in data loading, gradient computation, and communication for high-performance ML systems.
Candidates will design and maintain custom GPU kernels using Triton or CUDA to accelerate state-of-the-art machine learning workloads.
The position requires optimizing distributed training pipelines with PyTorch Distributed to maximize training throughput and step time efficiency.
Matching Summary
Match Score: 85
Motional Singapore is seeking a Senior ML Systems Engineer to enhance the performance and optimization of machine learning systems. The ideal candidate will have a strong background in software engineering, particularly with Python and PyTorch, and experience in optimizing machine learning workflows.
Skills & Requirements
Must-have
Proficiency in Python programming
Extensive PyTorch framework experience
GPU kernel development with CUDA or Triton
Distributed training pipeline optimization
Performance profiling using Nsight tools
Nice-to-have
Strong analytical problem-solving skills
Data-driven approach to technical challenges
Experience with model inference optimization
Key Requirements
Bachelor's, Master's, or PhD in Computer Science or related field