Ahsaas Bajaj
Lead Machine Learning Engineer at Instacart
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
- 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
Walmart Global Tech · Full-time
- Data ScientistFeb 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/
- Software Engineering InternMay 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
- Graduate Student ResearcherJan 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.
- Graduate Student Researcher - NGRAM NLP Lab (Prof. Mohit Iyyer)Jan 2020 - May 2020 · 5 mosAmherst, Massachusetts, United States
- Graduate Student Researcher - Isenberg School of ManagementSep 2019 - May 2020 · 9 mosAmherst, Massachusetts, United States
Education
- 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
- 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
At a glance
More about Ahsaas
Frequently asked
Explore more
Is this your profile?
Claim it to keep it up to date, or request removal. We're happy to help either way.
Similar roles



Similar profiles
- GPGermán PinedaPrincipal Software Engineer at Full-time · 3 yrs 1 moUnited States
- TFThomas FuSoftware Engineer | TLM at Google Research · Full-timeUnited States
- MPMikka PinedaFreelance Software Engineer at Self-employedUnited States
- STStanley T.Senior Design & Research Operations Principal at Full-time · 3 yrs 5 mosUnited States
- VHVasant HonavarFaculty Member, Huck Institutes of the Life Sciences at 11 yrs 11 mosUnited States
Looking for your next role?
Chat with Clera and we'll introduce you to the right opportunities.
This profile is based on publicly available information. Ahsaas is not affiliated with or endorsed by Clera. Privacy policy.