Lead AI Engineer

About the Position

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

We Offer

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)
• Focus on your professional growth:
  • 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)

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