Senior Machine Learning Engineer - Singapore

Job summary

Senior Machine Learning Engineer at Thoughtworks will bring engineering rigor and creative problem-solving to one of AI's fastest-evolving domains.

Responsibilities

• Implement and tune advanced model optimization techniques such as post-training quantization, pruning, and knowledge distillation.
• Configure and optimize inference runtimes and serving frameworks.
• Enable high-throughput serving using continuous batching, KV caching, speculative decoding, and asynchronous scheduling.
• Apply kernel fusion strategies to reduce latency and memory overhead.
• Evaluate trade-offs across accuracy, throughput, latency, and GPU/accelerator utilization for different hardware footprints.

Qualifications

• Strong foundation in machine learning with expertise in inference optimization techniques.
• Hands-on experience with modern inference runtimes and compilers.
• Familiarity in deep learning frameworks with production-ready engineering practices.

Job description