Lead Python AI Solution Engineer

We are looking for a Lead Python AI Solution Engineer to build production AI and LLM solutions, shape architecture and deliver reliable, scalable systems. You will lead Python-based RAG and agent work, integrate APIs, and partner with client technical leadership to steer decisions and delivery.

Responsibilities

• Design and deliver production-ready AI and LLM applications in Python
• Develop and sustain RAG pipelines along with AI agent architectures
• Implement and connect APIs that enable AI-driven capabilities
• Deploy and operate AI systems across cloud environments
• Ensure secure, scalable integration across platforms and services
• Collaborate with client Technical Leadership to set project direction and architecture
• Evaluate technical trade-offs with strong analytical judgment and recommend solutions
• Prototype and prove AI concepts using the right frameworks and tools
• Support experimentation and model tracking throughout the development lifecycle
• Translate business requirements into dependable technical implementations

Requirements

• 5+ years of professional software engineering experience, including substantial hands-on delivery of production AI and LLM applications
• High proficiency with Python 3.11/3.12+ as your primary programming language
• Deep experience with LLM application and agent frameworks such as LangChain, LlamaIndex, and PydanticAI
• Working knowledge of Hugging Face Models/Datasets, sentence-transformers, and tokenizers
• Hands-on familiarity with Python project and dependency tooling such as uv, venv, and Poetry
• Solid command of code quality and development tools including Ruff, Pytest, and Jupyter
• Strong background with machine learning libraries such as NumPy, pandas/Polars, and scikit-learn
• Practical understanding of deep learning frameworks such as PyTorch and TensorFlow/Keras
• Proven track record implementing vector search and RAG infrastructure such as FAISS, Qdrant, Chroma, Weaviate, Milvus, or pgvector with PostgreSQL
• Hands-on experience building AI APIs and deployment components with FastAPI, Pydantic, and Uvicorn
• Demonstrated skills in containerization and orchestration using Docker and Kubernetes
• Experience working with data infrastructure technologies such as PostgreSQL, Redis, and Apache Kafka
• English proficiency at B2 level (Upper-Intermediate) or higher

Nice to have

• Experience applying TRL for LLM post-training
• Knowledge of DSPy for building programmatic LLM pipelines
• Familiarity with LiteLLM to unify access to multiple LLM APIs
• Experience with experiment tracking tools such as MLflow and Weights & Biases
• Portfolio of AI application UI work using Streamlit, Gradio, or FastAPI with React/Next.js/Angular 19+

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.