Machine Learning Engineer

BuildBros Innovations; Bengaluru; Hybrid; INR 800,000 - 1,800,000; Full time

Benefits offered by this job

Health insurance

Professional development budget

Conference attendance

Remote-friendly hours

Job summary

BuildBros Innovations in Bengaluru is seeking a Machine Learning Engineer with 1–3 years of experience to build and deploy AI models for disease detection, NLP features, and recommendations across our SaaS suite.

You will optimize latency and cost, deploy via REST APIs using FastAPI and TensorFlow Serving, and work with product teams to identify ML opportunities. This role offers remote/hybrid flexibility with health insurance and growth potential.

Qualifications

• Strong Python with TensorFlow or PyTorch.
• Production ML deployment experience (not just Kaggle notebooks).
• CNN, transformer, and transfer learning expertise.
• Cloud ML services experience (AWS SageMaker, GCP Vertex AI, or similar).

Responsibilities

• Research, develop, and deploy ML models for computer vision and NLP tasks.
• Optimize model latency, accuracy, and inference cost for production deployment.
• Deploy models as scalable REST APIs using FastAPI and TensorFlow Serving.
• Implement model monitoring, drift detection, and automated retraining pipelines.
• Collaborate with product teams to identify and scope ML opportunities.

Skills

Python

TensorFlow

PyTorch

Production deployment

CNN/Transformer/Transfer learning

Cloud ML services

Tools

FastAPI

Docker

TensorFlow Serving

AWS SageMaker

GCP Vertex AI

MLflow

Job description

Machine Learning Engineer

Experience 1–3 Years

Mode Remote / Hybrid (Bengaluru)

Openings 2

About This Role

Build and deploy AI models for production products — plant disease detection for Indian farmers, intelligent recommendations, and NLP-powered features across our SaaS suite.

A Day in the Life

You profile inference latency, ship a distilled vision model behind FastAPI, wire drift checks into the monitoring stack, and pair with product on a new NLP feature spec.

What You'll Do

• Research, develop, and deploy ML models for computer vision and NLP tasks
• Optimize model latency, accuracy, and inference cost for production deployment
• Deploy models as scalable REST APIs using FastAPI and TensorFlow Serving
• Implement model monitoring, drift detection, and automated retraining pipelines
• Collaborate with product teams to identify and scope ML opportunities

What We're Looking For

• Strong Python with TensorFlow or PyTorch
• Production ML deployment experience (not just Kaggle notebooks)
• CNN, transformer, and transfer learning expertise
• Cloud ML services experience (AWS SageMaker, GCP Vertex AI, or similar)

Nice to Have

• LLM fine-tuning for domain tasks

Tools You’ll Use

Python TensorFlow PyTorch MLflow FastAPI Docker AWS SageMaker

₹8L – ₹18L CTC based on experience and publications

GPU compute budget for research and experimentation

Conference attendance and paper publication support

Remote-first with flexible hours for research work

Health insurance and professional development budget

Work on AI products that directly impact Indian farmers and communities

Your Growth Path

Senior ML Engineer (1-2 years) →

AI Lead (2-3 years) →

Head of AI

Interview Process

Total process: ~7–10 days

1. 1 Resume + Research Review 48 hours Review publications, Kaggle profile, GitHub ML projects.
2. 2 ML System Design 60 minutes Design an end-to-end ML system for a given use case.
3. 3 Coding + Model Building 90 minutes Build and optimize a model on a provided dataset.
4. 4 Research Discussion 45 minutes Discuss recent papers, model architectures, and your ML philosophy.
5. 5 Founder Chat 30 minutes Vision, culture, compensation.