Lead AI Software Engineer

EPAM Systems; México; Remote; MXN 900,000 - 1,300,000; Full time

Benefits offered by this job

Healthcare benefits

Paid time off

Upskilling & certification

Global career opportunities

Community involvement

Job summary

EPAM Systems Inc. seeks a Lead AI Software Engineer in Mexico to design, implement, and maintain scalable Python microservices with FastAPI for automated content generation and marketing workflows.

You will optimize diffusion models, connect pipelines to custom evaluation harnesses, and deploy containerized services on AWS/GCP using Infrastructure as Code. Collaborate with senior architects in a fast-moving Kanban environment to ship reliable generative AI pipelines.

Qualifications

• 5+ years in software engineering with Python backend development.
• Production applications using generative AI, LLMs, and diffusion models.
• Experience with asynchronous backends (e.g., FastAPI) and cloud deployment (AWS or GCP).
• Solid testing, evaluation harnesses, and MLOps principles in Python.

Responsibilities

• Design, implement, and maintain scalable Python microservices with FastAPI for automated content generation.
• Optimize and integrate diffusion models and multimodal AI tooling.
• Connect pipelines to Python-based evaluation harnesses to validate output quality.
• Deploy containerized workloads on cloud infrastructure with IaC tooling.
• Contribute to Kanban discussions and rapid feature iteration with senior leadership.

Skills

Python backend

FastAPI

AWS/GCP

CI/CD

Docker

MLOps

Asynchronous

Testing

Tools

OpenCV

FFmpeg

Terraform

Docker

Git

Job description

We are building a Lead AI Software Engineer role to deliver scalable Python microservices that automate content generation and marketing workflows. You will integrate generative and diffusion models, connect outputs to Python evaluation harnesses, and deploy containerized services on AWS in a fast-moving Kanban setup. Apply now to help ship reliable generative AI pipelines to production.ResponsibilitiesDesign, implement, and maintain scalable Python microservices with FastAPI for automated content and marketing material generationOptimize and integrate diffusion models and multimodal generative AI tooling, including image and video generation pipelinesConnect generation pipelines to custom Python-based evaluation harnesses to validate output quality prior to deploymentDeploy containerized workloads to cloud infrastructure using Infrastructure as CodeContribute to daily Kanban discussions, backlog grooming, and rapid feature iteration with senior architecture leadershipDeliver high-throughput generative AI content pipelines that reduce manual marketing asset creation timeEnsure new model versions integrate into the automated Python evaluation harness with zero regression in output qualityMaintain stable, containerized FastAPI microservices on cloud infrastructure with robust error handlingRequirementsProven track record of 5+ years in software engineering with deep expertise in Python backend developmentDemonstrated experience delivering production applications that use generative AI, large language models, and diffusion modelsHands-on experience with asynchronous backend frameworks such as FastAPI, plus cloud deployment on AWS or GCPSolid background in automated testing, evaluation harnesses, and MLOps principles in PythonStrong proficiency with Python, FastAPI, and AWS or GCPDeep understanding of Generative AI and diffusion models, including LLMs and image/video generationWorking knowledge of CI/CD, Docker, and GitNice to haveExperience with Adobe Firefly or commercial creative generation APIsFamiliarity with Terraform or other Infrastructure as Code toolsKnowledge of Computer Vision libraries such as OpenCV and FFmpegBackground in Media & Entertainment or video supply chainWe 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.