Senior ML Engineer
Primary stack
Job description
About the Position
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
About the Company
EPAM Systems is a global software engineering and product development company, delivering digital transformation and technology innovation to the world’s leading companies.
© EPAM. This job description was sourced from the employer's public career page. TheJob is not the employer — we index the posting and route candidates to the source. All content rights and hiring decisions belong to the employer.
EPAM helps organizations innovate their business processes and rethink the way they manage their businesses so they can remain competitive in this new digital age.
More at EPAM
All 2384 rolesLead Full-Stack Developer
EPAM · Argentina +1
Python Engineering Manager
EPAM · Mexico
Product Manager
EPAM · Argentina +2
Automation Tester (JavaScript)
EPAM · Brazil
