Senior Machine Learning Engineer
Core
Own production ML performance optimization initiatives from bottleneck identification through implementation, validation, and rollout.
Role type
Senior Machine Learning Engineer (Performance Optimization)
Builds
Optimized ML inference and training workloads for latency, throughput, and cost efficiency
Domain
Machine Learning Infrastructure / Systems Programming
Deliverable
production ML models
Required skills
Proficiency in Python, Profiling and benchmarking, Systems programming, Model execution optimization, Memory management, Compiler integration, GPU/kernel optimization, Distributed training optimization
Preferred skills
Experience in recommendation systems, ranking, search, or retrieval, PyTorch performance optimization, Custom CUDA/Triton kernel development, ML cost observability
Technologies
PyTorch, CUDA, Triton, GPU, Python
Responsibilities
Profile and diagnose bottlenecks across model execution, frameworks, runtimes, memory, and hardware; Implement improvements involving inference execution, batching, and custom operators; Support model launches through benchmarking and load testing; Build reusable performance benchmarks and tooling; Mentor engineers through collaboration and code reviews
