Lead Machine Learning Engineer

Job Summary

A company is looking for a Lead Machine Learning Engineer - AI Agents.

Key Responsibilities

• Architect and implement core infrastructure for agent-based LLM systems
• Train and fine-tune custom models to address specific challenges
• Lead collaboration across engineering, product, and research teams to align solutions with business impact

Required Qualifications

• Strong proficiency in Python and ML frameworks like PyTorch or TensorFlow
• Deep understanding of core ML concepts, including transformers and generative models
• Experience training large models on GPU clusters and integrating LLMs with external tools
• Hands-on experience with Retrieval-Augmented Generation (RAG) systems
• Proven ability to build end-to-end ML pipelines in a cloud-native environment