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

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)
• 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

Kindly be advised that the set of benefits, including learning, certification, and other opportunities, may vary depending on the role you apply for. Our recruiter will be able to share more details about the specific opportunity during your general interview.

EPAM strives to provide its global team of over 61,700 professionals in more than 55 countries with opportunities for professional growth from day one of collaboration. Our colleagues are the source of EPAM's success, so we value cooperation, strive to always understand our clients' business and aim for the highest quality standards. No matter where you are, you will join a dedicated, diverse community that will help you realize your potential to the fullest.