Lead Machine Learning Engineer

Thoughtworks is seeking a Lead Machine Learning Engineer to specialize in Inference Optimization. This role involves maximizing AI model efficiency, speed, and cost-effectiveness across various environments. The engineer will lead teams, design scalable inference systems, and mentor others, contributing to the development of internal frameworks and benchmarks for efficient inference.

Requirements

• Deep practical expertise in model and runtime optimization techniques
• Proven experience optimizing inference workloads using frameworks such as vLLM, NVIDIA Triton/Dynamo
• Strong proficiency in deep learning frameworks (PyTorch, TensorFlow)
• Ability to diagnose and optimize performance using profiling tools
• Solid understanding of GPU and accelerator architectures
• Experience designing and benchmarking scalable inference systems

Benefits

• Learning & Development opportunities
• Teamwork and mentorship