Lead AI Software Engineer

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.

Responsibilities

• Design, implement, and maintain scalable Python microservices with FastAPI for automated content and marketing material generation
• Optimize and integrate diffusion models and multimodal generative AI tooling, including image and video generation pipelines
• Connect generation pipelines to custom Python-based evaluation harnesses to validate output quality prior to deployment
• Deploy containerized workloads to cloud infrastructure using Infrastructure as Code
• Contribute to daily Kanban discussions, backlog grooming, and rapid feature iteration with senior architecture leadership
• Deliver high-throughput generative AI content pipelines that reduce manual marketing asset creation time
• Ensure new model versions integrate into the automated Python evaluation harness with zero regression in output quality
• Maintain stable, containerized FastAPI microservices on cloud infrastructure with robust error handling

Requirements

• Proven track record of 5+ years in software engineering with deep expertise in Python backend development
• Demonstrated experience delivering production applications that use generative AI, large language models, and diffusion models
• Hands-on experience with asynchronous backend frameworks such as FastAPI, plus cloud deployment on AWS or GCP
• Solid background in automated testing, evaluation harnesses, and MLOps principles in Python
• Strong proficiency with Python, FastAPI, and AWS or GCP
• Deep understanding of Generative AI and diffusion models, including LLMs and image/video generation
• Working knowledge of CI/CD, Docker, and Git

Nice to have

• Experience with Adobe Firefly or commercial creative generation APIs
• Familiarity with Terraform or other Infrastructure as Code tools
• Knowledge of Computer Vision libraries such as OpenCV and FFmpeg
• Background in Media & Entertainment or video supply chain

We offer

• 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

EPAM 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.

Remote in Mexico