Senior Machine Learning Engineer

SourcingXPress; Bengaluru; On-site; INR 4,000,000 - 5,000,000; Full time

Job summary

A technology startup in Bengaluru is seeking a Technical Lead to guide a team in building and deploying ML and GenAI products. Responsibilities include owning the technical roadmap, leading workflows for AI initiatives, managing model development, and mentoring team members. Candidates should have four years of experience, with strong skills in Python, as well as expertise in MLOps and relevant tools like PyTorch and HuggingFace. This role offers competitive compensation and opportunities for growth.

Qualifications

• At least 4 years of experience with one year in technical leadership.
• Strong understanding of Python and ML/DL concepts.
• Experience in MLOps and model deployment.

Responsibilities

• Own the technical roadmap for ML/GenAI initiatives.
• Lead workflow and system design for LLMs.
• Guide model development with code quality enforcement.
• Manage MLOps including CI/CD for models.
• Collaborate with teams for experiments and optimization.
• Mentor engineers and grow team capabilities.

Skills

Technical leadership

Python

Machine Learning (ML)

Deep Learning (DL)

Generative AI (GenAI)

MLOps

Tools

PyTorch

HuggingFace

LangChain

AWS

Neo4j

Milvus

Job description

Company Details

Hiredahead.com (Website: Visit Website). Business Type: Startup. Company Type: Product. Business Model: B2B. Funding Stage: Pre‑seed. Industry: Information Technology. Salary Range: ₹ 40‑50 Lacs PA.

Role Summary

Lead a small, high‑velocity team to design, build, and ship ML and GenAI products end‑to‑end, from problem framing and modeling to scalable deployment and monitoring.

Responsibilities

• Own the technical roadmap for ML/GenAI initiatives and align it with product goals and delivery timelines.
• Lead LLM workflow and agentic system design, including retrieval, tool‑use, and evaluation pipelines.
• Guide model development using PyTorch and HuggingFace; enforce code quality, testing, and model evaluation standards.
• Oversee MLOps: CI/CD for models, API design, containerization, orchestration, and cost/performance optimization on AWS.
• Collaborate with product, data, and infra teams; run experiments, A/B tests and error analysis.
• Mentor engineers, conduct code/model reviews, and grow team capabilities.

Must-Have Skills

• At least 4 years of experience with a minimum of one year in technical leadership leading projects or teams.
• General AI/ML Team Lead requirements: roadmap ownership, sprint planning, stakeholder communication, mentoring, and setting engineering best practices.
• Strong Python understanding: idiomatic code, testing, packaging, async I/O, performance profiling.
• Strong ML, DL, and GenAI understanding: supervised/unsupervised learning, transformer architectures, prompt engineering, evaluation, and safety/guardrails.
• Experience building LLM-based workflows and agents, including retrieval (RAG), tools, memory, and evaluation loops.
• Hands‑on experience with LangChain, HuggingFace, PyTorch, Neo4j, Milvus, and AWS for scalable, production‑grade systems.
• MLOps experience: model deployments, API development, observability, infra management (containers, orchestration, IaC), and data/version governance.

Preferred Skills

• Knowledge graph design and Neo4j schema modeling for semantic retrieval and reasoning.
• LLM fine‑tuning with deployment to production: LoRA/QLoRA, PEFT, quantization, evaluation, rollback strategies, and cost optimization.
• Startup experience: comfort with ambiguity, rapid iteration, and owning outcomes across the stack.