Computer Science Student @ IIT Delhi

Hi, I'm Arjun Mehta

Driven AI/ML enthusiast focused on building scalable machine learning systems and contributing to open source AI frameworks.

About Me

I am a driven and innovative computer science student at IIT Delhi with a strong passion for artificial intelligence, machine learning, and open source AI frameworks.

Experienced in contributing to AI-driven open source projects and developing scalable ML systems. I am eager to leverage my technical expertise and collaborative spirit in impactful roles focused on AI research, deployment, and optimization.

Education

Indian Institute of Technology (IIT) Delhi

May 2026

B.Tech in Computer Science and Engineering

CGPA: 9.3 / 10

Relevant Coursework

Machine Learning Deep Learning NLP Computer Vision DSA Operating Systems

Technical Skills

Languages

Python C++ Java R TypeScript

AI/ML Frameworks

TensorFlow PyTorch Scikit-learn HF Transformers

Cloud & DevOps

AWS (SageMaker) Azure ML GCP Docker Kubeflow MLflow

Databases

Pinecone (Vector) MongoDB MySQL

Other Tools

Git / GitHub / GitLab LangChain ONNX REST APIs Linux (Ubuntu) Jupyter

Experience

Machine Learning Intern

Microsoft India (Azure AI Team)

May 2024 - Jul 2024
  • Developed enhancements for Azure ML pipelines focusing on automated hyperparameter tuning.
  • Built custom dashboards for visualizing model performance metrics via Azure Monitor.
  • Optimized LLMs for low-latency inference on Azure Kubernetes Service with global teams.
  • Conducted a workshop on "Deploying Scalable ML Models with Azure" internally.

Open Source Contributor

Hugging Face & PyTorch

Sep 2023 - Present
  • Submitted 8+ PRs to Hugging Face Transformers, enhancing LLM optimization.
  • Improved PyTorch documentation for distributed training.
  • Developed fine-tuning scripts for LLMs shared with community (150+ GitHub stars).
  • Collaborated on ONNX runtime optimizations for faster edge inference.

Featured Projects

Academic Project

AI-Powered Sentiment Analysis

Designed and deployed a scalable sentiment analysis system using BERT-based models. Achieved 95% accuracy on social media datasets.

  • Deployed on AWS SageMaker
  • Microservices architecture with Docker & K8s
  • MLflow for experiment tracking
BERT AWS Kubernetes
100+ Stars

Real-Time Object Detection

Built a real-time system using YOLOv8 and deployed it on Google Cloud Run. Integrated Pinecone for efficient contextual object vector search.

  • Deployed on Google Cloud Run
  • Automated pipelines with GitHub Actions & Kubeflow
  • Adopted by IIT Delhi AI Club for workshops
YOLOv8 GCP Pinecone

Achievements

  • Google Summer of Code 2024

    Contributor with PyTorch (Distributed Training Enhancements).

  • 1st Place, IIT Delhi AI Hackathon 2024

    Developed an AI-powered chatbot for educational content delivery.

  • IIT Delhi Merit Scholarship

    Awarded in consecutive years (2023, 2024) for academic excellence.

Extracurriculars

  • Lead Organizer, IIT Delhi AI Club

    Organized 2 hackathons and mentored 30+ students in AI/ML projects.

  • Speaker, AI Community Days 2025

    Presented on "Optimizing Large Language Models for Edge Devices."

  • Mentor, GirlCodeIt Summer 2024

    Guided students on building AI apps and contributing to open source.