Sanjit Jain profile photo 1 on Tal

Sanjit Jain

Machine Learning Engineer || ML Platforms || ML Infra

Bengaluru

5+ years of work experience

Glance
Apr 20251 yr 5 mosPresent

Bengaluru, Karnataka, India

ShareChat
Sep 20231 yr 3 mosNov 2024

Bengaluru, Karnataka, India

Realtime Ranking and Personalization · Working on building real-time ranking capabilities for user-feed personalization on Moj and Ads. · Responsible for cost-reduction of 30% by moving to spot instances for training workflows. · Reduce training GPU cost by implementing mixed float operations and loss scaling.
ShareChat
Oct 20212 yrsSep 2023

Bengaluru, Karnataka, India

Moj Feed AI, Recommender Systems, Notification Personalisation and Volume Optimisation · Building infrastructure for personalised notifications capable of handling scale of millions of users. · Ran A/B experiments on personalised notifications as well as volume optimisation to achieve CTR gains and re-activation of users. · Achieved 80% savings in compute and SQL costs through various optimisations on infrastructure and storage. · Worked on solving Post Cold start problem for Moj. · Achieved a 0.X% retention gain in D1 users. · Increased the number of posts served to users by 4x, leading to improved diversity and post exploration. · Built early stage candidate generators targeting early post discovery to maximise chances of success. · Improved personalisation of posts in early stage to push for quick post embedding convergence. · Bridging the supply-demand gap in smaller Indian languages like Gujarati, Bhojpuri, Rajasthani, etc. through specific candidate genarators catering to these business usecases. · Running and analysing various A/B Experimentations of user and post variants to test different feed recommendation logic to optimise for metrics like user retention, video play time and engagement metrics. · Responsible for multiple core feed microservices and training pipelines and enabling autoscaling as well as assuring complete uptime of our feed models. Mentors - Hastagiri Vanchinathan, Director(Feed AI/Ranking), Michal Porvaznik, Staff ML Engineer and Jan Ruziev, Staff ML Engineer.
ShareChat
Apr 20216 mosSep 2021

Bengaluru, Karnataka, India

Recommender Systems, Feed AI, Duplicate Detection, Google Cloud TPUs · Worked in the modelling team of Moj Feed AI. · Worked on online recommender system leading to significant gains in business metrics like retention, video playtime, feed quality etc. · Built scalable, robust core and cost efficient modelling pipelines for our Feed AI Ranking component. · Worked on model inference, serving, load testing and deployment/monitoring for our core feed models. · Worked on Large scale nearest neighbour search engines for Feed Ranking and Content Understanding. · Extensively involved in developing and running multiple A/B Test Experiments on User and Post Level business metrics like retention, playtime, engagements etc. · Responsible for various end-to-end machine learning pipelines for Sharechat and Moj. Mentors - Hastagiri Vanchinathan, Director(Feed AI/Ranking) and Peddakota Vikash, Lead Data Scientist
Sparrosense
May 20193 mosJul 2019

Gurgaon, Haryana, India

Decoupling Localization and Classification in Single Shot Temporal Action Detection - Working on developing a robust model performs action localization and classification using a single end-to-end trainable procedure. Mentors - Ankit Agarwal, CEO and Ravikant Bhargav, Research Head
Algorithm Labs
May 20183 mosJul 2018

Work from Home

- Creating a chatbot using Google DialogFlow API to bridge the gap between buyers and sellers in the shipping industry. - Developing a model that Dynamically prices Hotels for Maximum Revenue. - Worked on developing the idea of Supply Chain Management. Mentors - Gaurav Sharma, CEO
Defence Research and Development Organisation (DRDO)
May 20183 mosJul 2018

Bangalore

Worked on a project on Rapid 3D Reconstruction of UAV video stream for Centre for Artificial Intelligence and Robotics(CAIR), a lab of DRDO. Mentor - Dr. Malay Nema, Scientist at DRDO

Sanjit’s AI Stack

claude code
claude code
deepseek
deepseek
hugging face
hugging face
kimi
kimi
ollama
ollama
vLLM
vLLM

Education

The LNM Institute of Information Technology

Bachelor of Technology - BTech, Computer Science

The LNM Institute of Information Technology

20164 yrs2020

12th standard

modern vidya niketan

20115 yrs2016

Join Sanjit and 10k+ other professionals on

Tal Boss

Tal is the app where bosses meet engineers & builders directly, without middlemen

Download Tal Boss

Sanjit Jain on Tal

Headline

Machine Learning Engineer || ML Platforms || ML Infra

About

- Data Scientist at Sharechat, India's leading in-house social media platform and short video-app. - Thorough knowledge and hands-on experience in Machine Learning, Recommender Systems, Nearest Neighbours Search Algorithms, Deep Learning and Multimodal Learning. - BTech. in Computer Science from LNM Institute of Technology.

Location

Bengaluru

AI stack

claude code

deepseek

hugging face

kimi

ollama

vLLM

Work experience

Machine Learning Engineer 3 - ML Platforms at Glance, Apr 2025 - Present · 1 yr 5 mos, Bengaluru, Karnataka, India

Lead Machine Learning Engineer at ShareChat, Sep 2023 - Nov 2024 · 1 yr 3 mos, Bengaluru, Karnataka, India, Realtime Ranking and Personalization · Working on building real-time ranking capabilities for user-feed personalization on Moj and Ads. · Responsible for cost-reduction of 30% by moving to spot instances for training workflows. · Reduce training GPU cost by implementing mixed float operations and loss scaling.

Data Scientist-2 at ShareChat, Oct 2021 - Sep 2023 · 2 yrs, Bengaluru, Karnataka, India, Moj Feed AI, Recommender Systems, Notification Personalisation and Volume Optimisation · Building infrastructure for personalised notifications capable of handling scale of millions of users. · Ran A/B experiments on personalised notifications as well as volume optimisation to achieve CTR gains and re-activation of users. · Achieved 80% savings in compute and SQL costs through various optimisations on infrastructure and storage. · Worked on solving Post Cold start problem for Moj. · Achieved a 0.X% retention gain in D1 users. · Increased the number of posts served to users by 4x, leading to improved diversity and post exploration. · Built early stage candidate generators targeting early post discovery to maximise chances of success. · Improved personalisation of posts in early stage to push for quick post embedding convergence. · Bridging the supply-demand gap in smaller Indian languages like Gujarati, Bhojpuri, Rajasthani, etc. through specific candidate genarators catering to these business usecases. · Running and analysing various A/B Experimentations of user and post variants to test different feed recommendation logic to optimise for metrics like user retention, video play time and engagement metrics. · Responsible for multiple core feed microservices and training pipelines and enabling autoscaling as well as assuring complete uptime of our feed models. Mentors - Hastagiri Vanchinathan, Director(Feed AI/Ranking), Michal Porvaznik, Staff ML Engineer and Jan Ruziev, Staff ML Engineer.

Data Scientist-I at ShareChat, Apr 2021 - Sep 2021 · 6 mos, Bengaluru, Karnataka, India, Recommender Systems, Feed AI, Duplicate Detection, Google Cloud TPUs · Worked in the modelling team of Moj Feed AI. · Worked on online recommender system leading to significant gains in business metrics like retention, video playtime, feed quality etc. · Built scalable, robust core and cost efficient modelling pipelines for our Feed AI Ranking component. · Worked on model inference, serving, load testing and deployment/monitoring for our core feed models. · Worked on Large scale nearest neighbour search engines for Feed Ranking and Content Understanding. · Extensively involved in developing and running multiple A/B Test Experiments on User and Post Level business metrics like retention, playtime, engagements etc. · Responsible for various end-to-end machine learning pipelines for Sharechat and Moj. Mentors - Hastagiri Vanchinathan, Director(Feed AI/Ranking) and Peddakota Vikash, Lead Data Scientist

Deep Learning Intern at Sparrosense, May 2019 - Jul 2019 · 3 mos, Gurgaon, Haryana, India, Decoupling Localization and Classification in Single Shot Temporal Action Detection - Working on developing a robust model performs action localization and classification using a single end-to-end trainable procedure. Mentors - Ankit Agarwal, CEO and Ravikant Bhargav, Research Head

Artificial Intelligence and Data Science Intern at Algorithm Labs, May 2018 - Jul 2018 · 3 mos, Work from Home, - Creating a chatbot using Google DialogFlow API to bridge the gap between buyers and sellers in the shipping industry. - Developing a model that Dynamically prices Hotels for Maximum Revenue. - Worked on developing the idea of Supply Chain Management. Mentors - Gaurav Sharma, CEO

Research Intern at Defence Research and Development Organisation (DRDO), May 2018 - Jul 2018 · 3 mos, Bangalore, Worked on a project on Rapid 3D Reconstruction of UAV video stream for Centre for Artificial Intelligence and Robotics(CAIR), a lab of DRDO. Mentor - Dr. Malay Nema, Scientist at DRDO

Education

Bachelor of Technology - BTech, Computer Science, The LNM Institute of Information Technology

12th standard, modern vidya niketan