Machine Learning Engineer Lead

NextMantra AI
Gurugram, Haryana, India

Position: Machine Learning Lead

Location: Bangalore or Coimbatore

Experience Required: 7+ years in AI/ML with a focus on custom model development and

deployment

Role Overview:

The Machine Learning Lead will oversee the full lifecycle of machine learning projects, from

concept to production deployment. The role requires strong technical expertise and

leadership to guide teams in delivering impactful AI-driven solutions.

Key Responsibilities:

● Design and implement scalable ML models for various business applications.

● Manage end-to-end ML pipelines, including data preprocessing, model training, and

deployment.

● Fine-tune foundation models and create small language models for deployment on

AWS Inferentia and Trainium.

● Deploy ML models into production environments using platforms such as AWS.

● Lead the implementation of custom model development, ML pipelines, fine-tuning,

and performance monitoring.

● Collaborate with cross-functional teams to identify and prioritize AI/ML use cases.

Required Skills:

● Expertise in ML frameworks like TensorFlow, PyTorch, and Scikit-learn.

● Strong experience deploying ML models on cloud platforms, especially AWS.

● In-depth knowledge of SageMaker and SageMaker Pipelines.

● Familiarity with RAG-based architectures, agentic AI solutions, Inferentia, and

Trainium.

● Advanced programming skills in Python with experience in APIs and microservices.

● Exceptional problem-solving abilities and a passion for innovation.

Position: AI/ML Engineer

Location: Bangalore or Coimbatore

Experience Required: 2-4 years in AI/ML development

Role Overview:

We are looking for a passionate AI/ML Engineer to contribute to the development of

AI-powered applications. The role involves building and deploying machine learning models,

ensuring performance optimization, and integrating solutions into production environments.

Key Responsibilities:

● Develop and train AI/ML models using frameworks like TensorFlow or PyTorch.

● Implement generative AI solutions such as RAG, agentic AI systems, document

extraction, custom model deployment, summarization, and text generation.

● Create APIs and microservices to integrate models with applications.

● Preprocess and clean data for high-quality model inputs.

● Monitor and troubleshoot deployed models to maintain performance.

● Collaborate with cross-functional teams to enhance AI workflows.

● Stay updated on the latest advancements in generative AI and integrate them into

projects.

Required Skills:

● Solid foundation in ML algorithms and concepts.

● Proficiency in GenAI frameworks like Amazon Bedrock ,Agentcore SageMaker,

Hugging Face,LangChain, Lang graph and Strands Agent.

● Proficiency in Python and libraries like NumPy, Pandas, and Scikit-learn.

● Familiarity with RAG frameworks, Agentic AI solutions, and generative AI.

● Strong collaboration skills and ability to work in a fast-paced environment.

Score my resume against this job, free →

Get your ATS score for this role — free. Score my resume free →