Lead AI Engineer

We are seeking an AI/ML Consultant to sit at the heart of our AI agent pipeline. The consultant will design and implement the core AI/ML components powering the event data extraction agent, including the RAG-based conversational assistant (UC2) and the reporting/analytics engine (UC3). The work spans LLM orchestration on Amazon Bedrock, knowledge base design, NLP pipelines and integration testing across the full agentic workflow.

Responsibilities

• Design and build the RAG (Retrieval-Augmented Generation) pipeline for UC2 — a conversational agent surfacing approved training content in the flow of work
• Develop UC3 analytics support — reporting and insights generation from event data
• Configure and fine-tune LLM inference on Amazon Bedrock (LLM Backbone)
• Build Natural Language Processing pipelines for parsing event input from File, Text and Voice modalities
• Design knowledge base architecture (vector storage, chunking strategy, retrieval optimization)
• Lead integration testing across the AI agent and downstream CRM output
• Collaborate with a Data Analytics consultant on ETL pipelines feeding the AI models
• Support HITL (Human-in-the-Loop) validation checkpoints for AI-generated outputs

Requirements

• Expertise in Amazon Bedrock, including LLM model selection, configuration and prompt engineering
• Proficiency in RAG architecture, covering vector databases, knowledge base design, embedding models and chunking strategies
• Skills in Natural Language Processing (NLP) such as entity extraction, classification and summarization
• Background in Python as the primary development language for ML pipelines
• Knowledge of AWS Lambda and Amazon S3 for serverless ML inference, event processing and data storage of training content and knowledge base artifacts
• Capability to perform integration testing for multi-step AI agent workflows
• Familiarity with Veeva CRM data schemas
• Upper-Intermediate English language proficiency (B2)

Nice to have

• Flexibility to use MLflow or SageMaker for experiment tracking and model versioning
• Experience with voice-to-text processing pipelines
• AWS Certified Machine Learning – Specialty

Benefits

With us you can:

• Work on a flexible schedule remotely or from any of our comfortable offices or coworking spaces in Ukraine
• Receive the necessary equipment to perform your work tasks
• Change projects and technology stacks within EPAM
• Gain experience in various business domains (Insurance, E-commerce, Healthcare, Finance, Travelling, Media, Artificial Intelligence, and more)
• Relocation opportunities may be available for eligible candidates, depending on the role and openings at other EPAM locations
• Participate in volunteer, charity programs and communities (both technical and interest-based)

We focus on your professional growth:

• You can plan your individual career path together with your manager
• Receive regular feedback from colleagues
• Improve your English for free with certified teachers (Speaking Clubs, client interview preparation courses, etc.)
• Get the opportunity to undergo free training and certification in AWS, GCP, or Azure Clouds
• Use the internal E-learn training program (18,200+ specialized training and mentoring programs)
• Access corporate accounts on LinkedIn Learning, Get Abstract and other partner resources
• Study at EPAM Solution Architecture School with the instructors who are practicing architects
• Develop as a leader, join Delivery Management, Resource Management, Leadership Essentials school and more
• Participate in internal communities (500+ meetups, technical discussions, brainstorming sessions, online events and conferences annually)

What we offer:

• Vacation and sick leave (including a sick leave without a medical certificate)
• A wide range of Voluntary Medical Insurance programs providing both medical treatment and various preventive options (including sports activities)
• Medical insurance for family members at corporate rates
• Company support during significant life events (childbirth or adoption, marriage, etc.)
• Support for psychological comfort: discounts on services from mental health specialists or coaches, thematic training
• E-kids program - a free programming language training program for EPAMers' children