ab Arjun Basandrai

Résumé

Arjun Basandrai
Machine Learning Engineer

Experience

AI Applications Engineer Intern

Micron Semiconductor Technology India · Ahmedabad, India
Feb 2026 - Jul 2026
  • Developed and deployed an Automatic Defect Classification system for wire bonders using a ResNet-50 image classifier, reducing defect identification and reject-code assignment time from 3-5 minutes to about 30 seconds.
  • Developed operator and admin dashboards for production use, adding automated inference, role-based access, API-key management, rate limiting, model versioning, prediction logging, and model monitoring.
  • Implemented a Gold Wire Monitoring System for 400 wire bonders that detects abnormal 99.9%-pure gold wire consumption, possible theft, material loss, and machine-wear signals and reduces machine-level analysis time from 15 minutes to 1 minute.
  • Implemented an Airflow ETL pipeline and SPC-based detection using control limits with hysteresis, CUSUM excursion, EWMA max-slope, and intra-lot variance checks.

Technologies: Python, PyTorch, FastAPI, PostgreSQL, Redis, React, TypeScript, DVC, Apache Airflow, Docker, Prometheus, Grafana, Microsoft Entra ID, OpenShift, Kubernetes

Software Engineer Intern

TraceLink Inc. · Pune, India
May 2025 - Oct 2025
  • Developed Solution Tester, a no-code UI regression-testing framework for TraceLink's OPUS platform that validates configurable customer pages through reusable BDD scenarios.
  • Designed the Playwright-based test engine and custom BDD parser, converting human-readable specifications into executable browser tests and integrating critical platform paths into the nightly regression suite.

Technologies: JavaScript, Playwright, BDD

Machine Learning Intern

Talent Recruit Software Pvt. Ltd. · Remote
Mar 2024 - Jul 2024
  • Researched and benchmarked bias-detection and mitigation methods for applicant-tracking and resume-screening systems.
  • Co-developed a bias-governance platform that enabled internal teams and clients to evaluate disparate-impact and other fairness metrics, supporting product changes that improved the disparate impact ratio for underrepresented groups by 19.7%.

Technologies: Python, Flask, IBM AI Fairness 360, Pandas, NumPy

Education

Master of Engineering in Computer Science

Cornell University · Ithaca, NY
Expected May 2027

Bachelor of Technology in Computer Science and Engineering

Vellore Institute of Technology · Vellore, India
May 2026
CGPA: 9.38/10

Higher Secondary Certificate Examination

Kundan Vidya Mandir Sr. Sec. School · Ludhiana, India
May 2022
Percentage: 96.6%

Projects & Research

Adaptive Resampling-based Training for Imbalanced Classification

arXiv | GitHub | PyPI
Mar 2025 - Feb 2026 under review at MLWA
  • Developed ART, an adaptive resampling method that updates training distributions using class-wise validation macro-F1 and empirical class priors.
  • Benchmarked ART against 11 class-imbalance methods across five tabular, image, and text datasets using 20 fixed-seed runs per method-dataset pair, improving macro-F1 by 2.64 percentage points on average across the tabular datasets.
  • Validated results using Wilcoxon signed-rank tests at p < 0.05 and released ART as a reusable PyTorch package.

Technologies: Python, PyTorch, scikit-learn, NumPy, Pandas, Optuna, pytest

TCGA GBM Survival Analysis

GitHub
Jan 2025 - Mar 2025
  • Built a survival-analysis pipeline using TCGA glioblastoma clinical and somatic-mutation data to study mutation patterns associated with patient outcomes.
  • Combined PyTorch autoencoders, K-means clustering, chi-square tests, KEGG pathway enrichment, Cox proportional-hazards modeling, Kaplan-Meier analysis, and log-rank tests.
  • Identified six candidate survival-associated genes and built an inference tool that generates individualized survival-probability curves from their mutation status.

Technologies: Python, PyTorch, scikit-learn, GSEApy, SciPy, Pandas, NumPy, Matplotlib, Seaborn

Elixir Chess Engine

GitHub | Lichess | CCRL 40/15 | CCRL Blitz
Feb 2024 - Nov 2024
  • Built a UCI-compatible C++ chess engine combining NNUE evaluation with alpha-beta search, quiescence search, transposition tables, and advanced pruning, achieving 3,500+ Elo on CCRL 40/15 and Blitz lists.
  • Trained and integrated a sparse 768-input NNUE model that improved playing strength by about 500 Elo, optimized inference with AVX2/AVX-512 SIMD and Lazy SMP multithreading, and validated changes through SPRT testing.
  • Won the Young Talents CEDR computer-chess tournament with Elixir 3.0 JA.

Technologies: C++ 20, Python, NNUE, SIMD, AVX2, AVX-512, Make

Birdly: Bird Species Identifier

GitHub (Model) | GitHub (Backend)
Nov 2022 - Jun 2023
  • Built an EfficientNetV2L image-classification system covering all 1,250 bird species found in India, trained on more than one million images.
  • Converted the dataset to TFRecords and trained on TPU v3, reducing training time from four hours per epoch on an NVIDIA P100 GPU to 19 minutes and achieving 82.3% top-1 test accuracy.
  • Built a Flask inference API that returns confidence-ranked top-k species predictions to a React frontend.

Technologies: Python, TensorFlow, Keras, Flask, React, Selenium, OpenCV, Pillow, NumPy

Leadership & Writing

Senior Core Committee Member

CodeChef-VIT Student Chapter
Mar 2023 - May 2026
  • Supported DevSoc 2023 and DevSoc 2024, each with 1200+ participants, by coordinating participant communication, event logistics, judging, and promotional campaigns.
  • Supported contestant onboarding, event-day operations, and communication for CookOff 8.0, a competitive programming contest with 500+ participants.
  • Co-led a Foundations of Machine Learning session, teaching supervised and unsupervised learning, regression, classification, and decision trees through guided code-along demonstrations.

Skills

Languages
Python C++ JavaScript TypeScript R SQL Java
Machine learning & Data Science
PyTorch TensorFlow Keras scikit-learn OpenCV Optuna Label Studio NumPy Pandas
Applications
FastAPI Flask React Streamlit PostgreSQL Redis SQLAlchemy Alembic
MLOps
DVC Apache Airflow Docker Kubernetes OpenShift Prometheus Grafana Git Linux pytest