Gen AI / ML Engineer
Yellow.ai · Bangalore
This job has been flagged as a Not American
Pay
$15k–235k
Setting
On-site
IH
Instahyre
Loading...
Find your dream job with Instahyre
Unlock limitless career opportunities with lakhs of job openings in 10,000+ top companies.
INVITE FRIENDS
FIND JOBS
LOGIN
SIGNUP
EMPLOYERS
yellow.ai (Formerly Yellow Messenger) at a glance
Founded in 2016
500 - 1000
Employee reviews
Overall
3.7
Work-Life Balance
3.7
Salary and Benefits
3.5
Company Culture
3.8
Career Growth
3.6
Diversity and Inclusion
3.8
Will Recommend to Friends
3.4
Gen AI / ML Engineer
yellow.ai (Formerly Yellow Messenger)
Bangalore4-8 Years
Apply to yellow.ai (Formerly Yellow Messenger)
Apply to yellow.ai (Formerly Yellow Messenger)
This job is no longer accepting applications
About yellow.ai (Formerly Yellow Messenger)
yellow.ai brings the best of AI+human-led conversational automation for enterprises of great repute like Schlumberger, Domino's, Dr. Reddy's Lab, PepsiCo, Bajaj Group, Indigo, Cipla, Siemens, MG Motors, and more. We have offices in 6 countries and have clients across 27 countries.
We're a team of 500+ makers, who've shipped over 650+ Intelligent Virtual Assistants. Our Virtual Assistants converse in over 120 languages, and our platform handles more than a billion conversations every month - over 50+ channels in text and voice! We're also named Leading conversational AI Platform, Distinguished CX Vendor, and Advanced Virtual Assistant Provider by Gartner. They're really critical and meet 1000+ Conversational AI Platforms. We're thrilled to be recognized by them!
We're one of the fastest growing SaaS leaders emerging from Asia and are backed on this journey with more than $100M in funding so far by partners like Lightspeed, Sapphire Ventures, WestBridge Capital and Salesforce Ventures. We also got honored recently, being one of the top 10 companies to work at by #LinkedinTopStartups' and a Great Place to Work at' certification.
Job Description
Function: Data Science and Analysis → Data Science / Machine Learning
Machine Learning
LLMs
Python
SQL
NLP
Generative AI
+6 more
PyTorch
Computer Vision
Transformers
AWS
Azure
Google Cloud (GCP)
We're looking for a driven and experienced Machine Learning Engineer to lead the design, development, and deployment of our next-generation AI systems. This role is for a hands-on builder with a strong background in large language models (LLMs), agentic systems, and end-to-end machine learning pipelines. You'll work at the intersection of research and production, turning prototypes into scalable, real-world solutions.
Responsibilities:
Design & Build Agentic Systems: Lead the development of LLM-powered agents and tools from proof-of-concept to production. You'll focus on agent design, including tool usage, planning, and memory systems, to create intelligent, autonomous workflows.
Develop & Deploy Production Features: Architect, implement, and deploy production-grade ML/NLP models for various use cases, including text classification, entity recognition, and complex data analysis. You will own the full lifecycle, from training and fine-tuning to real-time inference and monitoring.
Experiment & Iterate: Rapidly build and test different retrieval-augmented generation (RAG) strategies as embeddings, indexing, and re-ranking meet quality and latency targets. Your focus will be on delivering high-performance, production-ready features.
Build Scalable Infrastructure: Work across the stack to build and optimise data pipelines and serving infrastructure. You will be responsible for creating efficient systems for large-scale data processing and for low-latency, high-throughput model inference.
Ensure Quality & Performance: Develop robust evaluation frameworks and benchmark suites to systematically test model behaviours and ensure accuracy and reliability at scale. Continuously monitor production performance and refine models based on real-world feedback.
Requirements:
Experience: 4+ years of experience in ML engineering with a strong focus on LLMs, agent design, or NLP.
Technical Skills: Strong proficiency in Python and modern ML/NLP libraries (e. g., PyTorch, Hugging Face Transformers). Hands-on experience with the LLM/RAG stack, including vector databases and various retrieval strategies.
Agentic Systems: Demonstrable experience building LLM-based agents, including tool usage, planning, and memory systems.
Production Experience: Proven track record of deploying ML models and agents to production environments (AWS, GCP, or similar), with a strong understanding of APIs, microservices, and data pipelines.
Prototyping and Delivery: A track record of moving from an idea to a production-quality feature quickly, and clearly documenting results and trade-offs.
Data Engineering: Strong data engineering background with experience in frameworks like Apache Spark or Airflow.
Evaluation: Experience designing and implementing evaluation frameworks, metrics, and pipelines to validate model outputs at scale.
Nice-to-Have:
Experience with model fine-tuning and training workflows (LoRA, PEFT, etc. ).
Familiarity with LLM orchestration frameworks like Agno, CrewAI, or AutoGen.
Proven contributions to open-source projects related to LLMs or AI infrastructure.
Full-stack fluency (e. g., Next.js/React) for building and deploying features end-to-end.
Benefits at yellow.ai (Formerly Yellow Messenger)
Competitive Salary • Medical insurance • remote friendly • WFH setup allowance • monthly wifi stipend • Flexible vacation policy including parental leaves
yellow.ai (Formerly Yellow Messenger)'s Tech Stack
Node.js
Golang
React.js
Vue.js
Job posted by
Akasmat Pradhan
Founder & CEO
at Magna Hire
Similar jobs
AI / ML Engineer
Maieutic Semiconductors
Bangalore
AI / ML Engineer
Nightfall
Bangalore
Lead LLM Engineer
Opkey
Bangalore
Lead Data Scientist
SatSure
Bangalore
AI / ML Engineer
ACL Digital
Bangalore
Lead AI / ML Engineer
Oolka
Bangalore
AI / ML Engineer
Noora Health
Bangalore
AI / ML Engineer
StockGro
Bangalore
Senior ML Engineer
Matters
Bangalore
Team Lead - AI / ML
Oolka
Bangalore
Jobs by Location
Jobs in Bangalore
Jobs in Delhi / NCR
Jobs in Hyderabad
Jobs in Mumbai
Jobs by Function
Software Engineering Jobs
Marketing Jobs
Sales Jobs
Internship Jobs
For Employers
Post Your Jobs
Success Stories
Product Academy
Resources
Instahyre
Workscape
About
Privacy
Terms
Connect
Help Center
Contact Us
Listed by Yellow.ai for a position based in the United States. Employers on this board attest they are hiring domestically.