Rohan Jain

Software Engineer

San Francisco, California, United States16 yrs 10 mos experience

AI ML PractitionerAI Enabled

Key Highlights

Led Tensorflow product development at Google.
Developed large-scale ETL systems for Knowledge Graph.
Implemented AdWords features generating significant revenue.

Stackforce AI infers this person is a Backend-focused Software Engineer with expertise in Machine Learning and Data Processing.

Contact

Skills

Core Skills

Machine LearningSystem DesignInformation RetrievalBig Data ProcessingData MiningAlgorithmsProgramming

Other Skills

TensorflowETLAPI DevelopmentData ProcessingTeam LeadershipData AnalysisC++Natural Language ProcessingArtificial IntelligencePythonComputer ScienceData StructuresMapReduceMatlabSQL

About

Software engineer focused on backend development, both infrastructure and quality. Experience ranges from early stage prototyping and product development to implementation, leading teams and supporting critical production systems, gained during stints with flagship google products such as Tensorflow, Google Search and AdWords products. Keen interest in data mining, machine learning, large scale systems and distributed systems. Specialties: System design, Information Retrieval, Big data processing, Machine Learning, Data Mining, Natural Language Processing, Programming, Advanced Algorithms

Experience

16 yrs 10 mos

Total Experience

4 yrs 3 mos

Average Tenure

4 mos

Current Experience

Google deepmind

Senior Staff Software Engineer

Dec 2025 – Present · 4 mos · Mountain View, California, United States · On-site

Google

Senior Staff Software Engineer

Aug 2010 – Jan 2026 · 15 yrs 5 mos · Mountain View

Feb 2020 - Present: Tech lead and manager for the team (~20 engineers) developing the core Tensorflow product.
July 2016 - Feb 2020: IC on the Tensorflow team working on tf.data, remote execution and other features
June 2013 - June 2016: Knowledge Graph team. Led efforts such as
Developing third party API's for external data and service providers to contribute to Google
Building systems to fetch, crawl and store external data required to build the graph
Building a massive scale ETL system to process and ingest data into the graph
Efforts enabled multiple new google search experiences for things such as recipes, concerts, streaming radio stations, movie streaming, gas prices, food ordering services etc. affecting XXm+ queries per day
May 2012 - May 2013: AdWords advertiser suggestions team
Launched new geo targeting recommendations for more than XXXk+ advertiser campaigns
August 2010 - April 2012: AdWords backend team
Launched 2 AdWords quality improvements that bring in $Xb+ in yearly incremental revenue

TensorflowETLAPI DevelopmentData ProcessingTeam LeadershipMachine Learning+1

Microsoft - bing, machine learning div

SDE Intern

Jun 2009 – Sep 2009 · 3 mos · Redmond

Worked on problems such as discovering semantically similar queries, building a dictionary of items from a small seed set of items etc.

Ibm india research lab, machine learning div

Intern

Jun 2008 – Aug 2008 · 2 mos · New Delhi Area, India

Detected faulty machines in a cluster using machine learning techniques monitoring > 10k variables.

Tryst - 2008, iit delhi's technical festival

Organizing Secretary

Aug 2007 – Mar 2008 · 7 mos

Led a team of 350 students to ensure effective running of IIT Delhi's largest ever technical festival, with participation increasing to four times previous years