Senior AI/ML Engineer with experience in GenAI

The vacancy is open for remote or office-based work within Latvia.

4000- 5000 EUR Gross salary range for Employment contract in Latvia.

Client

Our client is a top legal recruiting firm creating a data-driven platform that unites news, analytics, real-time case tracking, profiles, salary data, and events - powered by AI for actionable insights.

Project overview

This platform combines news, analytics, real-time tracking of deals and cases, interconnected profiles of firms and lawyers, salary information, event schedules, and more - all enhanced with AI to provide valuable, actionable insights.

Position overview

We are looking for a talented Senior AI/ML Engineer to join our team and help build advanced NLP and GenAI-powered features on our platform. You will work closely with product and data teams to develop intelligent solutions including entity recognition and linking, AI-powered content summarization, and scalable ML pipelines on AWS.

Responsibilities

• Develop and deploy AI/ML models leveraging GenAI (OpenAI API) for natural language understanding, summarization, and insights extraction
• Build named entity recognition (NER) and entity linking solutions tailored to legal domain data (law firms, cases, deals, people)
• Implement scalable NLP pipelines for processing news, legal documents, and transaction data from multiple sources
• Design, train, and evaluate ML models to improve search, classification, and recommendation features
• Collaborate with AWS teams to deploy and maintain models using SageMaker, manage datasets on S3, and ensure reliable operation and scalability
• Integrate AI capabilities seamlessly into the web platform, working alongside front-end and backend engineers
• Continuously research new AI models and NLP techniques relevant to legal data and user experience

Requirements

• Strong experience with GenAI technologies, especially OpenAI’s API for advanced NLP tasks
• Proven expertise in NER and entity linking techniques, preferably on domain-specific data
• Solid background in NLP workflows (tokenization, embedding, text classification, summarization)
• Comfortable with Python ML frameworks (TensorFlow, PyTorch, Hugging Face Transformers)
• Hands-on experience deploying ML models and workflows on AWS (SageMaker, S3 storage, Lambda, etc.)
• Familiarity with legal domain data or interest in legal tech is a strong plus
• Good software engineering practices and version control (Git)
• Excellent problem-solving skills and ability to work cross-functionally
• Strong communication skills and eagerness to learn emerging AI/ML technologies

Nice to have

• Experience with knowledge graphs or graph databases for entity linking
• Prior work on real-time data ingestion and analytics pipelines
• Experience with containerization (Docker, Kubernetes)