AB

Ahsaas Bajaj

Lead Machine Learning Engineer at Instacart

United States500+ connectionsLinkedInUpdated 10 months ago

About Ahsaas

Ahsaas Bajaj is a Senior Machine Learning Engineer, currently at the forefront of large-scale real-world recommender systems at Instacart. He specializes in natural language processing, algorithms, and data structures. His skill-set involves a strong command of ML tools and libraries, including AWS SageMaker, MLflow, TensorFlow, Airflow, and Snowflake.

His work got called out in Instacart's Q4 2024 shareholder letter, quoting, "Further improved order quality by upgrading our ML-powered item replacement model. Training on 10X more data and using large language models for richer signals led to higher customer retention by improving replacement satisfaction."

Before his stint at Instacart, Ahsaas worked with the search and personalization team at Walmart Labs as a Data Scientist, where he skillfully implemented large-scale recommendation systems that greatly enhanced the Walmart Grocery Experience. In summer 2020, he interned with the Document Insights Team at Instabase Inc., where he made noteworthy strides in improving the capabilities of their NLP pipeline.

A graduate of the University of Massachusetts Amherst, Ahsaas dived deep into the fields of Natural Language Processing (NLP), Information Retrieval, and Deep Learning during his studies in Computer Science. His dedicated research in these areas led to the submission of two research papers at the prestigious 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP).

Experience

  1. 3 yrs 5 mos

    • Senior Machine Learning Engineer II (L6)
      Mar 2025 - Present · 6 mosCurrent
    • Senior Machine Learning Engineer (L5)
      Sep 2023 - Mar 2025 · 1 yr 7 mos
    • Machine Learning Engineer II (L4)
      Apr 2022 - Sep 2023 · 1 yr 6 mos
  2. Walmart Global Tech · Full-time

    • Data Scientist
      Feb 2021 - Apr 2022 · 1 yr 3 mosSunnyvale, California, United States

      • Member of Search & Personalization Team • Build large scale recommendation systems to power Walmart Grocery Experience • Led the efforts on customer's repurchase journey and product type affinities • Tech Stack: PySpark, Hive, Scala, SQL, Hadoop, Google Cloud Platform • Patent - https://patents.google.com/patent/US20230244741A1/

  3. Instabase · Full-time

    • Software Engineering Intern
      May 2020 - Aug 2020 · 4 mosSan Francisco, California, United States

      • Applied ML and NLP techniques for document understanding (structured, unstructured data) • Improved recall by 17% for extraction of person & company names from client datasets of w2 forms and resumes • Delivered BERT-based entity recognition and Table Question Answering in production using GRPC & Kubernetes • Developed NLP framework to run on company infrastructure that allows model training, fine-tuning & evaluation • Incorporated use-cases like NER (structured data) and Question-Answering (natural language data) for real-world datasets

  4. Goldman Sachs

    • Graduate Student Researcher
      Jan 2020 - May 2020 · 5 mos

      • Worked with a team of four research scientists towards ‘Preserving critical facts in abstractive summarization’ • Built scalable abstractive summarization models that preserve critical facts in legal and financial documents • Conceptualized novel summarizer by leveraging pre-trained language models to build sentence-level classifiers • This was an industrial mentorship program in collaboration with Information Extraction and Synthesis Laboratory (IESL) at UMass Amherst.

  5. 9 mos

    • Graduate Student Researcher - NGRAM NLP Lab (Prof. Mohit Iyyer)
      Jan 2020 - May 2020 · 5 mosAmherst, Massachusetts, United States
    • Graduate Student Researcher - Isenberg School of Management
      Sep 2019 - May 2020 · 9 mosAmherst, Massachusetts, United States

Education

  1. University of Massachusetts AmherstMaster's degree, Computer Science2019 - 2020

    Grade: CGPA: 4.00 (on scale of 4) Research: • Prof. Andrew Mccallum's IESL Lab: An Instance Level Approach for Shallow Semantic Parsing in Scientific Procedural Text (EMNLP 2020 Paper) • Prof. Mohit Iyyer's NGRAM NLP Lab: Probing tasks for language models • Industrial mentorship (Goldman Sachs CoreAI): Preserving critical facts in abstractive summarization Coursework: • Intro Natural Language Processing by Dr. Mohit Iyyer • Neural Networks: A Modern Introduction by Dr. Erik Miller • Automated Knowledge Base Construction by Dr. Andrew McCallum • Machine Learning by Dr. Justin Domke • Algorithms for Data Science • Information Retrieval

  2. Netaji Subhas Institute of TechnologyBachelor of Engineering - BE, Electronics and Communications Engineering2013 - 2017

    Grade: 85.78% Activities and societies: NSIT Programming Group, Computer Society of India Silver Medalist for batch of 2013-2017. Department Rank: 2/180

Skills

Other

Machine LearningMatlabLaTeXJavaArtificial IntelligenceImage ProcessingObject-Oriented Programming (OOP)Android DevelopmentImage AnalysisResearchData ScienceData AnalyticsCProgrammingSQLC++PythonData StructuresMySQLgRPCKubernetesGoogle CloudOpenCVAlgorithmsAWSSageMakerMLflowTensorFlowAirflowSnowflake

At a glance

Experience6 years
Currently at3 yrs 5 mos
Based inUnited States
Studied atUniversity of Massachusetts Amherst
Roles8
Skills30
LanguagesHindi · English
LinkedInView profile

More about Ahsaas

Frequently asked

Is this your profile?

Claim it to keep it up to date, or request removal. We're happy to help either way.

Claim or remove

Similar roles

Browse all jobs

Similar profiles

Find more professionals

Looking for your next role?

Chat with Clera and we'll introduce you to the right opportunities.

Chat with Clera

This profile is based on publicly available information. Ahsaas is not affiliated with or endorsed by Clera. Privacy policy.