Lead AI Engineer

EPAM Systems; Colombia; On-site; COP 120,000,000 - 180,000,000; Full time

Benefits offered by this job

Healthcare benefits

Paid time off

Job summary

EPAM Systems seeks a Lead AI Engineer to manage ML systems from concept to production. You will design scalable architectures, build robust pipelines, and collaborate with product and business teams to define success metrics.

The role emphasizes leadership in ML lifecycle, handling large-scale data, and deploying models including recommendations and forecasting. Strong MLOps, PySpark, and PyTorch skills are required.

Qualifications

• 5+ years of experience in machine learning engineering or a related discipline.
• At least 1 year of leadership experience.
• Strong expertise in MLOps practices for overseeing the ML lifecycle.
• Experience with batch and real-time data processing architectures.

Responsibilities

• Manage the complete ML systems lifecycle from problem formulation to deployment and monitoring.
• Design scalable ML architectures and robust data pipelines (batch and real-time).
• Build and launch models such as recommender, ranking, forecasting, and classification/regression systems.
• Collaborate with cross-functional teams to define KPIs and success metrics.

Skills

Leadership experience

ML lifecycle

Cross-team collaboration

Problem solving

Education

B.S./M.S. in Computer Science or related

Tools

PySpark

Python

Kafka

MLflow

PyTorch

Job description

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.ResponsibilitiesManage the complete ML systems lifecycle, covering problem formulation, data collection, feature engineering, model training, evaluation, deployment, and monitoringCreate and implement scalable ML architectures, including both batch and real-time pipelinesHandle large-scale structured and unstructured datasets, such as clickstream, transactional, text, and image dataBuild and launch models like recommendation systems, ranking models, forecasting models, and classification/regression systemsUtilize deep learning methods, including neural networks, transformers, and representation learning, as neededEstablish and maintain feature stores, model registries, and experiment tracking systemsGuarantee production readiness by optimizing for latency, scalability, reliability, and cost efficiencyPartner with engineering, product, and business teams to establish KPIs and success metricsRequirementsOver 5 years of experience in machine learning engineering or a related disciplineAt least 1 year of applicable leadership experienceStrong expertise in MLOps practices for overseeing the ML lifecycleSkilled in PySpark and Python for developing scalable data and ML pipelinesProven background in designing and launching production-grade ML systemsSolid understanding of batch and real-time data processing architecturesExperience handling large-scale structured and unstructured datasetsEnglish language skills at a B2 level or aboveNice to haveExperience with Apache Kafka and MLflowKnowledge of machine learning concepts and PyTorchWe offerInternational projects with top brandsWork with global teams of highly skilled, diverse peersHealthcare benefitsEmployee financial programsPaid time off and sick leaveUpskilling, reskilling and certification coursesUnlimited access to the LinkedIn Learning library and 22,000+ coursesGlobal career opportunitiesVolunteer and community involvement opportunitiesEPAM Employee GroupsAward-winning culture recognized by Glassdoor, Newsweek and LinkedInEPAM is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, national origin, age, sexual orientation, gender identity or expression, disability, protected veteran status, or any other characteristic protected by applicable law.