AI/ML Architect

DataArt is seeking a Lead AI/ML Architect for a legal recruiting platform that aggregates data from hundreds of public sources to create a unified ecosystem of structured legal data. The role involves developing and deploying AI/ML models, leading AI architecture, technical strategy, and team mentorship for a data-driven platform serving lawyers and law firms.

• 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

• Strong experience designing and implementing end-to-end AI/ML solutions in production environments
• Hands-on experience with GenAI, including practical usage of the OpenAI API
• Solid background in NLP, with hands-on experience in Named Entity Recognition (NER) and entity linking
• Hands-on experience working with AWS, including services such as SageMaker and S3
• Strong Python skills with practical experience using TensorFlow, PyTorch, and Hugging Face Transformers
• Practical experience designing and implementing RAG (Retrieval-Augmented Generation) solutions
• Proven experience building and implementing Recommendation Engines
• Nice to have: Experience architecting scalable AI/ML systems on Amazon Web Services, experience with complex data-intensive machine learning pipelines, experience combining NLP, GenAI, and recommendation systems