Mohit Jain — Software Engineer
Staff Software Engineer with 7.5+ years of experience building and scaling large-scale ML/AI platforms and distributed systems across Dream11, Swiggy, and Amazon. Currently at Atlan, excited to build developer-first data platforms at global scale.Previously a founding member of Darwin, Dream11’s end-to-end ML platform powering the organization’s ML lifecycle. I designed and scaled multiple high-impact systems, including a Feature Store handling 200M+ RPM (p95 ~3ms), a compute platform running 4K+ Spark/Ray clusters and 5K+ daily workflows, low-latency ML serving at 60M+ RPM, and a spends platform enabling 200M+ daily discounts across 900+ campaigns.Beyond core platforms, I’ve built AI agents for Spark anomaly recovery, infra incident triage, and data scientist assistance. I also optimized Spark using Celeborn and dynamic configurations, reducing infrastructure costs by ~50%.I’ve led org-wide reliability initiatives, helped scale engineering teams, and partnered closely with data science, analytics, and product teams to accelerate ML adoption. I enjoy turning complex data and ML challenges into high-performance, reliable, and developer-friendly platforms.Speaker at ApacheCon China 2025.Passionate about distributed systems, ML infrastructure, AI agents, and building platforms that help teams innovate faster.
Stackforce AI infers this person is a SaaS expert specializing in scalable ML infrastructure and distributed systems.
Location: Mumbai, India
Experience: 10 yrs 7 mos
Skills
- Machine Learning Infrastructure
- Distributed Systems
- Data Visualization
Career Highlights
- Expert in building scalable ML/AI platforms.
- Reduced infrastructure costs by ~50% through optimization.
- Speaker at ApacheCon China 2025.
Work Experience
Atlan
Staff Software Engineer (2 mos)
Dream11
Staff Software Engineer (6 mos)
Software Development Engineer III (2 yrs)
Software Development Engineer II (2 yrs 1 mo)
Swiggy
Software Development Engineer II (1 yr 4 mos)
Software Development Engineer (1 yr 8 mos)
Amazon
Software Development Intern (5 mos)
BITS Pilani, Hyderabad Campus
Training And Development Coordinator, Placement Division (1 yr)
Assistant Placement Coordinator,Placement Division (1 yr 2 mos)
Logistics Coordinator, Placement Division (4 mos)
SAG Labs, Defence Research and Development Organisation
Intern (2 mos)
HCL Infosystems Ltd
Intern (2 mos)
Education
Bachelor of Engineering (B.E.) at Birla Institute of Technology and Science, Pilani
Master of Science (M.Sc.) at Birla Institute of Technology and Science, Pilani
High School at Subodh Public School,Jaipur