Senior AI Engineer

We are looking for a Senior AI Engineer to own the end-to-end lifecycle of machine learning systems, from problem formulation to production deployment. You will design scalable ML architectures, build robust pipelines, and collaborate across teams to deliver impactful AI-driven solutions.

Responsibilities

• Own end-to-end lifecycle of ML systems: problem formulation, data collection, feature engineering, model training, evaluation, deployment, and monitoring
• Design and implement scalable ML architectures including batch and real-time pipelines
• Work with large-scale structured and unstructured datasets such as clickstream, transactional, text, and image data
• Develop and deploy models such as recommendation systems, ranking models, forecasting models, and classification/regression systems
• Apply deep learning techniques including neural networks, transformers, and representation learning where appropriate
• Build and maintain feature stores, model registries, and experiment tracking systems
• Ensure production readiness through latency optimization, scalability, reliability, and cost efficiency
• Collaborate with engineering, product, and business teams to define KPIs and success metrics

Requirements

• 3+ years of experience in machine learning engineering or a related field
• Expertise in MLOps practices for managing the ML lifecycle
• Proficiency in PySpark and Python for building scalable data and ML pipelines
• Background in designing and deploying production-grade ML systems
• Understanding of batch and real-time data processing architectures
• Experience working with large-scale structured and unstructured datasets
• English proficiency at B2 level or higher

Nice to have

• Familiarity with Apache Kafka and MLflow
• Knowledge of Machine Learning 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