Default profile banner
SK

Sagar Kaushik

@sagarkaushik

Machine Learning Engineer at Comcast

Delhi, India

linkedin.com/in/sagar-kaushik

ComcastBirla Institute of Technology and Science (BITS)

Sagar Kaushik is an experienced Machine Learning Engineer with expertise in Deep Learning, NLP, and Computer Vision. He has a strong background in developing and deploying advanced models, including recommendation systems and LLMs. His experience includes optimizing models for production environments and conducting rigorous A/B testing to drive business improvements.

Experience

Machine Learning Engineer

Comcast

Full-timeJul 2022 - PresentChennai, India

• Trained and integrated a CBOW-inspired recommendation model into the service, achieving 2x the offline recall compared to the popularity heuristic. • Saved nearly $15,000 annually by fine-tuning the model daily and eliminating the indexing API before inference. • Created click-based and inverse propensity score weighted popularity models to rank the rows on pages for cold-start users. • Conducted A/B testing for thousands of users; the weighted model showed 7% better engagement and ranking metrics. • Developed a two-tower model with pairwise learning-to-rank for ranking rows on a page using embeddings of user watch history and rows with their metadata, and achieving 0.72 NDCG. • A/B tested the metadata-enriched row-ranking model, resulting in a 12% increase in successful sessions. • Devised a GPT2 based LLM trained on user watch history data to predict the next watched program, and used that for ranking programs with 0.8 recall @ 10.

Machine Learning Intern

Skit.ai

InternshipJan 2022 - Jun 2022Bangalore, India

• Built a convolutional recurrent neural network based noisy speech classifier which had 73.5% test accuracy. • Trained a transformer model to convert graphemes to phonemes with 6.8% PER, followed by a Text-to-Speech model on the obtained phonemes, and then used a pre-trained HifiGAN vocoder to generate audios.

Machine Learning Intern

Samsung R&D Institute

InternshipJul 2021 - Dec 2021Bangalore, India

• Implemented Federated Learning with Differential Privacy to train neural networks for predicting user demographics, obtaining 63% accuracy for age and 76% for gender. • Computed the cost of federated learning and simulated training on various data distribution scenarios on AWS.

Data Science Intern

PASS Consulting Group

InternshipMay 2019 - Jul 2019Hyderabad, India

• Built a neural network to predict water flow rates for different combinations of reservoir tank valve states with 67% R2.

Education

Birla Institute of Technology and Science (BITS)

Dual Degree

Electrical & Electronics, Physics

Aug 2017 - Jun 2022Grade: CGPA: 8.15/10

Skills

Python
SQL
PySpark (Apache Spark)
Pandas
TensorFlow
Keras
PyTorch
PyTorch Lightning
Huggingface Transformers
Scikit-learn
Auto-Sklearn
Recommender Systems
NLP
Computer Vision
Deep Learning
Statistics
Private AI
Databricks
MLflow
Weights & Biases
AB Testing
Scripting
Git
AWS
Docker
Kubernetes