SDE II - ML

Overview

Build core machine learning systems for an AI shopping agent, including low-latency inference platforms, evaluation frameworks, and production services for experimental models.

What you'll do

• Build and scale recommendation inference, vector search, feature store, and real-time data systems.
• Develop offline and online evaluation frameworks for ML model releases.
• Create regression tests and latency and cost tracking tools.
• Prototype engineering ideas for AI shopping features.
• Turn promising prototypes into scalable platform components.
• Optimize throughput and serving latency across CPU and GPU clusters.
• Collaborate with applied scientists and backend engineers to productionize ML models.

What you'll need

• 3+ years of software engineering experience.
• 2+ years focused on ML infrastructure, model serving, or recommendation systems.
• BTech or MTech in Computer Science, Machine Learning, or a related technical field.
• Strong Python or Java coding skills and distributed systems experience.
• Experience with model serving tools such as Triton, vLLM, or TorchServe.
• Experience with vector databases, feature stores, and databases.
• Experience deploying and monitoring high-throughput ML systems in production.

Details

• Location: Bangalore.