Weekday AI

Machine Learning Engineer

Weekday AI · Bengaluru, Karnataka, India

Technology, Information and Internet · 11-50 employees

7 h ago
Principal (10+ yrs) Full-time India
Log in to apply, save this posting, or score it against your profile with AI.

About the role

Build and operate production infrastructure to transform ML research into scalable, low-latency AI services and agentic workflows. Develop MLOps pipelines, optimize model inference, and ensure the reliability and observability of AI systems.

What they look for

Machine Learning Engineering Python MLOps Kubernetes Docker REST APIs Distributed Systems CI/CD LLM Applications LangChain LlamaIndex Kafka Spark Airflow Kubeflow Vector Databases

Requirements

Requires 5-11+ years of experience in ML Engineering or Backend Engineering with expertise in Python and distributed systems. Must have hands-on experience with LLM tooling, containerization, and cloud-native deployment on AWS.

Full description

This role is for one of the Weekday's clients

Min Experience: 5+ years

Location: Bengaluru

JobType: full-time

We are looking for a Machine Learning Engineer to build and operate the production infrastructure that transforms machine learning research into scalable, reliable, and low-latency AI services. You will partner closely with Applied Science, Product, and Platform Engineering teams to operationalize machine learning models, LLM-powered applications, and agentic workflows that power real-world enterprise products.

This role focuses on building production-ready ML systems, developing MLOps infrastructure, and ensuring AI services are secure, observable, cost-efficient, and highly available. You'll play a key role in enabling both traditional machine learning models and modern generative AI applications to move seamlessly from experimentation into production.

Key ResponsibilitiesProduction Machine Learning Systems• Convert prototype machine learning models into production-grade, scalable services with well-defined API interfaces.

  • Deploy and optimize models across various domains including predictive analytics, recommendation systems, forecasting, NLP, and generative AI.
  • Refactor, containerize, version, deploy, and continuously monitor machine learning models for production readiness.
  • Collaborate with Applied Science teams to improve model performance, scalability, and operational efficiency.

LLM & Agentic AI Infrastructure• Integrate AI applications with enterprise LLM gateways, model routing systems, and prompt management frameworks.

  • Support retrieval-augmented generation (RAG), vector search, and knowledge retrieval architectures.
  • Build and maintain agentic AI workflows, orchestration frameworks, and safe execution patterns.
  • Implement prompt versioning, experimentation, A/B testing, dynamic orchestration, and AI safety guardrails.

MLOps & Platform Engineering• Design and maintain CI/CD pipelines for machine learning models and AI services.

  • Build batch and streaming data pipelines using modern orchestration and distributed processing frameworks.
  • Develop online feature pipelines, feature stores, model registries, and experiment tracking infrastructure.
  • Automate model lifecycle management, deployment workflows, rollback strategies, and continuous delivery.

Microservices & Distributed Systems• Develop high-performance inference services using REST and gRPC APIs.

  • Build scalable microservices supporting low-latency online predictions.
  • Implement schema versioning, structured outputs, and API reliability standards.
  • Optimize service performance to consistently meet stringent latency and availability targets.

Monitoring, Reliability & Observability• Implement comprehensive monitoring across AI systems, including traces, logs, metrics, model performance, and infrastructure health.

  • Detect model drift, data quality issues, feature degradation, and operational anomalies.
  • Design resilient systems with autoscaling, caching, retries, circuit breakers, fallback mechanisms, and graceful degradation.
  • Track infrastructure utilization, latency, cost, and AI service quality through production dashboards.

Developer Experience & Enablement• Create reusable SDKs, templates, command-line tools, and deployment frameworks.

  • Build sandbox environments and documentation that simplify AI application development.
  • Collaborate with engineering teams to establish best practices for production ML, MLOps, and AI engineering.
  • Mentor engineers and contribute to improving platform standards and development workflows.

Required Qualifications• 5–11+ years of experience in Machine Learning Engineering, MLOps, Platform Engineering, or Backend Engineering supporting production ML systems.

  • Strong software engineering skills with expertise in Python and at least one of Java, Go, or Scala.
  • Solid understanding of distributed systems, concurrency, API design, testing, and scalable software architecture.
  • Experience deploying and operating production machine learning services.
  • Hands-on experience with orchestration frameworks and LLM tooling such as LangChain, LlamaIndex, OpenAI Function Calling, Agent frameworks, or similar technologies.
  • Knowledge of retrieval-augmented generation (RAG), vector databases, knowledge graphs, and AI agent architectures.
  • Experience building data pipelines using Airflow, Kubeflow, Spark, Flink, Kafka, or similar technologies.
  • Strong experience with Docker, Kubernetes, microservices, REST APIs, and gRPC services.
  • Familiarity with experiment tracking, model registries, feature stores, drift detection, A/B testing, and shadow deployments.
  • Experience implementing observability using tools such as OpenTelemetry, Prometheus, Grafana, or similar monitoring platforms.
  • Experience deploying cloud-native applications on AWS or comparable cloud environments.
  • Understanding of security best practices including RBAC, secrets management, audit logging, and PII protection.

Preferred Qualifications• Experience building enterprise AI platforms or large-scale MLOps infrastructure.

  • Knowledge of vector databases, retrieval systems, and knowledge graph technologies.
  • Experience supporting LLM-powered applications, AI agents, and autonomous workflows.
  • Familiarity with cloud cost optimization and multi-tenant SaaS architectures.
  • Strong understanding of production reliability engineering and distributed system design.

Ideal Candidate ProfileThe ideal candidate:

  • Thinks beyond models and focuses on delivering measurable business outcomes.
  • Prioritizes reliability, scalability, security, and operational excellence.
  • Enjoys designing production systems that balance performance, cost, and maintainability.
  • Works effectively across Applied Science, Product, and Engineering teams.
  • Believes in automation, developer productivity, and platform engineering best practices.
  • Documents processes clearly and enjoys mentoring other engineers.

Why Join Us?Join a team building next-generation AI infrastructure that enables enterprise-scale machine learning, LLM-powered applications, and intelligent automation. You'll help shape production AI platforms that power real-world products while working with modern MLOps technologies, distributed systems, and cutting-edge generative AI.

Must-Have Skills• Machine Learning Engineering

  • Python
  • MLOps
  • Kubernetes
  • Docker
  • REST APIs
  • Distributed Systems
  • CI/CD
  • LLM Applications

Good-to-Have Skills• Machine Learning

  • Python
  • LangChain
  • LlamaIndex
  • Kafka
  • Spark
  • Airflow
  • Kubeflow
  • MLflow
  • Vector Databases
  • Retrieval-Augmented Generation (RAG)
  • AWS