Mohit Jain

Background

About Mohit Jain

Staff Software Engineer with 7.5+ years of experience building large-scale ML/AI platforms and distributed systems at Dream11, Swiggy, and Amazon. Founding member of Darwin, Dream11’s ML platform powering the organization’s ML lifecycle. Designed and scaled: • Feature Store – 200M+ RPM, p95 = 3ms • Compute Platform – 4K+ Spark/Ray clusters, 5K+ workflows daily • MLServe – low-latency serving at 60M+ RPM • Spends Platform – 200M+ daily discounts, 900+ campaigns Beyond platforms, I’ve built AI agents for Spark anomaly recovery, incident triage, and DS assistance, while optimising Spark with Celeborn to cut infra costs by 50%. I also scaled our engineering team, led reliability initiatives cutting incidents, and shared learnings as an ApacheCon 2025 speaker. Passionate about distributed systems, ML infra, and AI agents, with a goal of transforming complex data and ML challenges into high-performance, reliable, and (developer+data science) friendly platforms. Built AI Agents for Spark anomaly recovery, infra incident triage, optimized Spark with Celeborn + dynamic configs, cutting infra costs by 50%. Experienced in scaling teams, driving org-wide reliability initiatives, and partnering with DS, analytics, and product teams to accelerate ML adoption. Speaker at ApacheCon China 2025. Passionate about distributed systems, ML infra, AI agents, and building platforms that help teams innovate faster.