Senior Machine Learning Engineer

Job Description

Archipelago is seeking a Senior Machine Learning Engineer to join their growing labs team in India to work on Gen AI systems. The ideal candidate will be comfortable with full-stack development, from building models that run in an extraction pipeline to UIs that test and evaluate the results of experiments. Archipelago iterates quickly, ships every day, builds for the long term, and is looking for smart, independent engineers who want to apply their trade with like-minded people.

This role involves designing and developing machine learning and deep learning systems, building RAG pipelines with LLMs, and building Agentic AI applications.

Responsibilities:

• Designing and developing machine learning and deep learning systems
• Building RAG pipelines with LLMs
• Building Agentic AI applications
• Design and implement tracking and evaluation tools to assess model performance and data accuracy, focusing on product relevance based on user interaction feedback
• Implementing appropriate ML algorithms
• Orchestration of long running jobs for data extraction
• Building UIs to test & evaluate results of ML experiments
• Translate research papers into high-quality, production-ready code
• Ship code to production frequently
• Coach and mentor more junior engineers

Requirements:

• 4+ years experience as a Machine Learning Engineer or similar role
• Familiarity working with LLMs, prompt engineering and RAG frameworks (llamaindex)
• Understanding of data structures, data modeling and software architecture
• Deep knowledge of math, probability, statistics and algorithms
• Ability to write robust code in Python
• Familiarity with machine learning frameworks (like Keras, PyTorch and XGBoost)
• Knowledge of how to deploy and maintain ML services in a production environment
• Experience in designing and deploying feature engineering pipelines in production
• Excellent communication skills
• Ability to work in a team
• Outstanding analytical and problem-solving skills

The role offers:

• Opportunity to work on cutting-edge GenAI systems
• Chance to build and implement machine learning and deep learning systems
• Be part of Archipelago’s India based AI team