Artificial Intelligence Engineer

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

Job summary

Hiredahead.com in Bengaluru is hiring a Lead ML/GenAI architect to design and ship end-to-end ML products. You will own the technical roadmap, guide LLM workflows, and drive production-grade systems across data, infra, and product teams.

The role demands strong Python, PyTorch, LangChain, and MLOps expertise, with experience in building scalable AI solutions on AWS. A startup mindset and collaborative leadership are essential.

Qualifications

• Minimum 4+ years in AI/ML with at least 1 year in technical leadership.
• Proven ability to own roadmaps and drive delivery in fast-paced settings.
• Strong Python coding, testing, and performance profiling skills.
• Experience with LLM workflows, RAG, tools, memory, and evaluation loops.
• Hands-on with LangChain, HuggingFace, PyTorch, and cloud infra on AWS.

Responsibilities

• Own the technical roadmap for ML/GenAI initiatives and align with product goals.
• Lead LLM workflow design, including retrieval, tooling, and evaluation pipelines.
• Guide model development with PyTorch and HuggingFace; enforce quality and tests.
• Oversee MLOps: CI/CD for models, API design, containers, orchestration on AWS.
• Collaborate with product, data, and infra teams; run experiments and analyses.
• Mentor engineers and conduct code/model reviews to grow team skills.

Skills

Leadership

Python

ML & GenAI

LLM workflows

MLOps

LangChain

HuggingFace

PyTorch

AWS

Problem solving

Tools

LangChain

HuggingFace

PyTorch

Neo4j

Milvus

AWS

Job description

Job Description

This is a Permanent Role with a Valued Clients of Hireahead.com

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

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.