Staff Machine Learning Engineer (9+)
wbd | 120 days ago | Bangalore

You will be part of a team focused on re-training, model hosting, cost optimization, and managing production workflows at scale.

  • Build and maintain pipelines for model fine-tuning and retraining, including LoRA-based workflows

  • Integrate and maintain vector search services and semantic similarity infrastructure

  • Design scalable model serving solutions for open-source and foundation models

  • Develop systems for experiment tracking, model versioning, and evaluation

  • Monitor production models for drift and performance degradation

  • Manage compute cost and resource optimization across distributed training jobs

  • Integrate Human-in-the-Loop (HITL) workflows and offline labeling into training pipelines

  • Support model deployment for varied model architectures, including Vision-Language Models, Convolutional Neural Nets, and Embedding Generation models

  • Stand up and maintain Feature Store and data versioning infrastructure

  • Architect and implement RAG pipelines for video metadata, summarization, and Q&A

  • Build evaluation frameworks to assess LLM performance, hallucination frequency, and structured response accuracy

 

What to Bring:

  • 9+ years of experience in machine learning engineering, with end-to-end ML workflow expertise

  • Strong background in model retraining, fine-tuning, and evaluation techniques

  • Experience deploying and managing open-source model servers (e.g., Triton, TorchServe, Ray Serve)

  • Proficient in managing cost-effective distributed computing environments (e.g., Kubernetes, Ray, SageMaker)

  • Familiar with experiment tracking tools (e.g., MLflow, Weights & Biases) and model versioning strategies

  • Deep understanding of ML domains including NLP, RecSys, and reinforcement learning

  • Experience with real-time inference systems and streaming data pipelines is a plus

  • Familiarity with labeling tools, HITL workflows, and offline data curation strategies

    • Comfort working in Agile development environments and collaborating across global teams

Official notification

⚡ Hot Jobs Trending Now

SRE
Sr. SRE Engineer
Stripe | Bangalore, India
DEV
Backend Developer
Coinbase | Remote, India
Infra
Cloud Infra Lead
Datadog | Pune, India
ML
MLOps Architect
Anthropic | Hyderabad
Data
Fivetran Data Eng.
Fivetran | Mumbai
SRE
Sr. SRE Engineer
Stripe | Bangalore, India
DEV
Backend Developer
Coinbase | Remote, India
Infra
Cloud Infra Lead
Datadog | Pune, India
ML
MLOps Architect
Anthropic | Hyderabad
Data
Fivetran Data Eng.
Fivetran | Mumbai
SDE
Staff Software Eng.
Airbnb | Gurgaon, India
Prod
Platform Engineer
Databricks | Bangalore
QA
Quality Assurance
GitLab | Remote
Security
Cloud Security
Zscaler | Mumbai
UX
Product Designer
Figma | Pune, India
SDE
Staff Software Eng.
Airbnb | Gurgaon, India
Prod
Platform Engineer
Databricks | Bangalore
QA
Quality Assurance
GitLab | Remote
Security
Cloud Security
Zscaler | Mumbai
UX
Product Designer
Figma | Pune, India
Contact US

Let's work laptop charging together

Any question or remark? just write us a message

Send a message

If you would like to discuss anything related to payment, account, licensing,
partnerships, or have pre-sales questions, you’re at the right place.