Bungee Tech | Internship
Machine Learning Engineer
Jun 2023 - Jul 2023

The project involved matching retail pet and grocery products to market conditions for effective pricing.

The project managed a substantial dataset of 5TB, featuring over 100 million products with comprehensive attributes such as titles, images, descriptions, sizes, flavours, model numbers, and UPCs.

Exploratory analysis was conducted on text and string-based data utilising Python libraries and SQL queries via AWS Athena.

A Transformer-based model architecture with Approximate KNN was developed, which was evaluated for its performance, achieving benchmark F1 scores of 96% against the WDC product matching benchmark and 91% against an internal evaluation dataset.

IIT Madras | Research Internship
Spatial Navigation Team
May 2024 - July 2024

Developed a system that uses Graph Database, Graph Neural Networks and Large Language Models(LLMs) to navigate a grid world, answer user queries, and provide relevant information about the environment.

Processed the dataset to store it in a Neo4j graph database and for other downstream tasks.

Used the LLM(Llama3) to question the graph database and generate Cypher queries for the questions asked by the user. The prompts were dynamically tuned using Langchain to improve query generation.

Implemented Q-learning method to explore the grid world. Used the explored data to query the graph neural network using the LLM(Llama3).

I presented this work in the 5th Annual CNSL Meet.