Mohit Jain

Software Engineer

Mumbai, India10 yrs 7 mos experience
AI EnabledAI ML Practitioner

Key Highlights

  • Expert in building scalable ML/AI platforms.
  • Reduced infrastructure costs by ~50% through optimization.
  • Speaker at ApacheCon China 2025.
Stackforce AI infers this person is a SaaS expert specializing in scalable ML infrastructure and distributed systems.

Contact

Skills

Core Skills

Machine Learning InfrastructureDistributed SystemsData Visualization

Other Skills

Feature StoreAI AgentsJavaApache SparkApache KafkaCore JavaCassandraApache FlinkPythonAmazon Web Services (AWS)SQLRayKubernetesAmazon Elastic MapReduce (EMR)Golang

About

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.

Experience

Atlan

Staff Software Engineer

Jan 2026Present · 2 mos · India · Remote

Dream11

3 roles

Staff Software Engineer

Promoted

Jul 2025Jan 2026 · 6 mos

  • Founded and scaled Darwin ML Platform; delivered Feature Store, Compute, MLServe, Spends Platform.
  • Built AI Agents (Spark anomaly detection, infra triage, CoT-based DS assistant).
  • Speaker at ApacheCon China 2025 (https://www.youtube.com/watch?v=Sx8mir55PCQ)

Software Development Engineer III

Jul 2023Jul 2025 · 2 yrs

JavaApache SparkApache KafkaCore JavaCassandraApache Flink+7

Software Development Engineer II

Jul 2021Aug 2023 · 2 yrs 1 mo

JavaApache SparkApache KafkaCore JavaCassandraApache Flink+6

Swiggy

2 roles

Software Development Engineer II

Mar 2020Jul 2021 · 1 yr 4 mos

  • Developed Rill Stream Platform (Flink + Kafka) → 700K events/sec.
  • Built Dipole caching layer in Golang + Redis → p99 = 3ms.
  • Built Compass - inhouse dashboard platform using (Druid + Superset)
  • Twice awarded Swiggy MVP.
JavaApache SparkApache KafkaCore JavaApache FlinkAmazon Elastic MapReduce (EMR)+5

Software Development Engineer

Jul 2018Mar 2020 · 1 yr 8 mos

JavaApache SparkApache KafkaCore JavaApache FlinkAmazon Elastic MapReduce (EMR)+5

Amazon

Software Development Intern

Jan 2018Jun 2018 · 5 mos · Hyderabad Area, India

PythonAmazon Web Services (AWS)SQL

Bits pilani, hyderabad campus

3 roles

Training And Development Coordinator, Placement Division

Jul 2016Jul 2017 · 1 yr · Hyderabad Area, India

Assistant Placement Coordinator,Placement Division

May 2015Jul 2016 · 1 yr 2 mos · Hyderabad Area, India

Logistics Coordinator, Placement Division

Jan 2015May 2015 · 4 mos · Hyderabad Area, India

Sag labs, defence research and development organisation

Intern

May 2015Jul 2015 · 2 mos · New Delhi Area, India

  • Zero Day Malware Detector using Heuristic Approach
  • Introduction to SOOT and JIMPLE for the de-assembling purpose of malware samples.
  • Implementation of NHGK (Neighbourhood Hash Graph Kernel) algorithm for handling call graph structures.
  • Advancement using Linear SVM(Support Vector Machine) introduction for robust results.

Hcl infosystems ltd

Intern

May 2014Jul 2014 · 2 mos · Jaipur Area, India

  • Development of an online Employee Appraisal System using Java Enterprise Edition
  • Integration of the portal with SQL server for data storage and management

Education

Birla Institute of Technology and Science, Pilani

Bachelor of Engineering (B.E.) — Computer Science

Jan 2013Jan 2018

Birla Institute of Technology and Science, Pilani

Master of Science (M.Sc.) — Economics

Jan 2013Jan 2018

Subodh Public School,Jaipur

High School

Jan 2012Present

Stackforce found 100+ more professionals with Machine Learning Infrastructure & Distributed Systems

Explore similar profiles based on matching skills and experience