Lead AI Software Engineer

EPAM is a leading global provider of digital platform engineering and development services. We are committed to having a positive impact on our customers, our employees, and our communities. We embrace a dynamic and inclusive culture. Here you will collaborate with multi-national teams, contribute to a myriad of innovative projects that deliver the most creative and cutting-edge solutions, and have an opportunity to continuously learn and grow. No matter where you are located, you will join a dedicated, creative, and diverse community that will help you discover your fullest potential.
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.

Work arrangement
Yes