Lead AI Engineer

We are seeking a Lead AI Engineer to manage machine learning systems from initial concept through to production launch. This role involves designing scalable ML architectures, building robust pipelines, and working with cross-functional teams to deliver effective AI solutions.

Responsibilities

• Manage the complete ML systems lifecycle, covering problem formulation, data collection, feature engineering, model training, evaluation, deployment, and monitoring
• Create and implement scalable ML architectures, including both batch and real-time pipelines
• Handle large-scale structured and unstructured datasets, such as clickstream, transactional, text, and image data
• Build and launch models like recommendation systems, ranking models, forecasting models, and classification/regression systems
• Utilize deep learning methods, including neural networks, transformers, and representation learning, as needed
• Establish and maintain feature stores, model registries, and experiment tracking systems
• Guarantee production readiness by optimizing for latency, scalability, reliability, and cost efficiency
• Partner with engineering, product, and business teams to establish KPIs and success metrics

Requirements

• Over 5 years of experience in machine learning engineering or a related discipline
• At least 1 year of applicable leadership experience
• Strong expertise in MLOps practices for overseeing the ML lifecycle
• Skilled in PySpark and Python for developing scalable data and ML pipelines
• Proven background in designing and launching production-grade ML systems
• Solid understanding of batch and real-time data processing architectures
• Experience handling large-scale structured and unstructured datasets
• English language skills at a B2 level or above

Nice to have

• Experience with Apache Kafka and MLflow
• Knowledge of machine learning concepts and PyTorch

Benefits

• International projects with top brands
• Work with global teams of highly skilled, diverse peers
• Healthcare benefits
• Employee financial programs
• Paid time off and sick leave
• Upskilling, reskilling and certification courses
• Unlimited access to the LinkedIn Learning library and 22,000+ courses
• Global career opportunities
• Volunteer and community involvement opportunities
• EPAM Employee Groups
• Award-winning culture recognized by Glassdoor, Newsweek and LinkedIn