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)