Senior ML Engineer

We are seeking a Senior ML Engineer to join a strategic initiative in the travel sector, focused on building a production-grade Knowledge Graph platform with integrated Retrieval-Augmented Generation (RAG) capabilities.

This role will drive the development of advanced semantic content discovery and relationship modeling solutions, leveraging cutting-edge cloud and ML technologies.

Responsibilities

• Design and implement LLM-based entity and relationship extraction workflows
• Develop prompt templates and orchestration logic for Graph RAG
• Build evaluation frameworks for precision/recall tracking and gold set validation
• Implement human-in-the-loop review workflows
• Optimize LLM usage for cost and performance
• Collaborate with cross-functional teams to align platform features with business goals
• Integrate vector embeddings and similarity search into platform architecture
• Ensure robust API development using REST or GraphQL
• Oversee ETL pipeline development for data ingestion and transformation

Requirements

• 4+ years of experience in ML/AI engineering
• Hands-on experience with LLM APIs such as OpenAI, Anthropic or AWS Bedrock
• Strong prompt engineering and LLM orchestration skills
• Proficiency in Python ML frameworks
• Skills in embedding models and vector similarity search
• Knowledge of AWS services including Neptune, Bedrock, OpenSearch, S3 or Lambda
• Experience with Infrastructure-as-Code using Terraform or CloudFormation
• Competency in CI/CD automation and Git-based version control
• Familiarity with Agile or Scrum project management methodologies

Nice to have

• Experience building RAG (Retrieval-Augmented Generation) systems
• Knowledge of graph neural networks or knowledge graph embeddings
• Background in NLP or information extraction
• Familiarity with LangChain, LlamaIndex or similar frameworks

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