Skip to content

Aakash Madabhushi

Machine Learning Engineering

I build the systems a research team runs on: the pipeline that feeds training, the harness that measures the result, and the service that puts it behind an API.

Most of my research work is infrastructure rather than modeling. In the SJSU robotics lab I own the path from raw teleoperation data to a scored policy: pre-training quality gates that refuse to hand a corrupted tensor to a training run, a PyTorch and CUDA pipeline with configurable augmentation and 5K-step checkpointing that can resume a 60K-step run after an interruption, Weights & Biases tracking, and a physical evaluation harness, built for a 100-trial protocol, that scores rollouts one at a time, returns the arm to start on its own, guards against stale caches, and logs each failure mode separately. The researchers get a number they can trust and a run they do not have to restart.

The modeling is real too, across four settings: a 52M-parameter Action Chunking Transformer policy scored on physical rollouts, a graph attention network classifying host-pair links for intrusion detection, a 1.57M-parameter audio-video emotion model built from scratch whose cross-attention fusion matched baselines 30 to 50 times its size, and classical models on tabular data. So is the serving end, where the model has to answer inside a latency budget rather than in a notebook: the threat-detection system returns a retrieval-grounded explanation in 0.87 s at the median, and one model runs behind a FastAPI service that rebuilds and redeploys itself on every commit. Underneath all of it I have built the primitives by hand, from a GPT-2 and a byte-pair tokenizer written from scratch to attention-speedup work and a Phi-3 fine-tune.

0.87 s
median end-to-end latency per retrieval-grounded explanation
92%
pick-and-place success, 50 physical rollouts at the trained cube position
0.90
F1 on 329 held-out host-pair links, 0.94 precision and 0.87 recall
49,633
demonstration frames through the quality gates and into training

8 projects in this area

Every link goes to code or data you can inspect yourself.

Four panels from the report, one per emotion class, naming the facial region the model attended to for each: mouth and jaw for angry, the full face for happy, diffuse for neutral, eyes and brow for sad.
ML Engineering

Reading Emotion from Voice and Face

Matched models 30 to 50 times its size on a four-class emotion benchmark with a 1.57M-parameter audio-video network trained from scratch, then showed which facial regions and frequency bands drove every prediction.

74.6%
validation accuracy, four-class CREMA-D on an actor-independent split
1.57M
parameters, 30 to 50× fewer than comparable multimodal baselines
12×
faster training epochs after caching decoded media, 1,100 s to 88 s
  • Python
  • PyTorch
  • CUDA
  • Cross-attention fusion
  • Grad-CAM
  • OpenCV
  • SpecAugment

SJSU deep learning course, team project · Feb to May 2026

ML EngineeringAI EngineeringData Engineering

Explainable Network Threat Detection

Condensed 27 million raw network connection logs into a map of which machines talked to each other, flagged the suspicious pairs with a graph neural network, and had a language model explain every alert in plain English.

27.1M
Zeek network flows aggregated into the graph
0.90
F1 on 329 held-out host-pair links, 0.94 precision and 0.87 recall
  • PyTorch Geometric
  • Airflow
  • dbt
  • Snowflake
  • Pinecone
  • +1 more

SJSU capstone, team project · Feb to May 2026

ML EngineeringSoftware Engineering

Thyroid Recurrence Predictor

A cancer-recurrence model taken the whole way to production: trained, wrapped in an API, containerized, and redeployed automatically on every commit.

  • Python
  • scikit-learn
  • FastAPI
  • Docker
  • GitHub Actions
  • +1 more

Personal project · Apr 2025

The SO-101 arm on the lab desk beside the marked cube it picks up and the clear container it places the cube into.
Robot LearningML Engineering

Teaching a Robot Arm to Pick and Place

Trained a real robot arm to pick up a cube and place it from 50 human demonstrations, then built a resumable evaluation harness and scored 50 physical rollouts at 92% success.

92%
pick-and-place success, 50 rollouts at the trained cube position
2.49 cm
average placement error on successful trials
  • Python
  • PyTorch
  • CUDA
  • LeRobot
  • ACT
  • +3 more

Graduate Research Assistant, SJSU Applied Data Science · Mar 2026 to present

AI EngineeringML Engineering

InvoSense: Invoice Processing with a Human Approval Gate

Reads a photo of an invoice, pulls out the amounts and dates, and scores its own confidence in each field. It refuses to record a payment until a person approves it.

100%
of payment actions blocked pending human review
  • Python
  • LangGraph
  • Groq
  • Llama 3.2 Vision
  • Llama 3.1
  • +2 more

Personal project · Jun 2026

Two charts from the report: a bar chart ranking which features predict delay, and a pie chart showing the top five predictors against all other factors.
Data EngineeringML Engineering

U.S. Flight Delays: Batch Analytics to Real-Time Prediction

Trained a delay predictor on 9.5 million U.S. flights, then ran it inside a live stream so each incoming flight gets a delay probability within seconds. Where a flight lands and when it leaves drove most of the signal.

9.5M
flight records processed, 15 monthly BTS files totalling 3.1 GB
0.711
F1 for the random forest on a stratified 20% test split, 79.1% accuracy
  • Python
  • PySpark
  • Spark MLlib
  • Spark SQL
  • Kafka
  • +3 more

SJSU big data course, team project · Dec 2025

Data EngineeringML Engineering

Live Wikipedia Edit Sentiment

Streams every English Wikipedia edit as it happens and scores the tone of each edit summary, so you can spot pages turning contentious in real time.

  • Python
  • Kafka
  • RoBERTa
  • Hugging Face

Personal project · Nov 2025

Tools I've shipped with here

Training & modeling

  • PyTorch
  • CUDA
  • PyTorch Geometric
  • Transformers
  • Cross-attention fusion
  • scikit-learn
  • pandas
  • Imitation learning
  • Fine-tuning

Research infrastructure

  • Weights & Biases
  • Hugging Face Hub
  • Checkpoint & resume
  • Mixed precision
  • Pre-training data validation
  • Stratified splits
  • Class weighting

Serving

  • FastAPI
  • Docker
  • GitHub Actions
  • AWS EC2
  • Streamlit

Relevant experience

The parts of my roles that bear on this work. The résumé has all of it.

Instructional Student Assistant, Machine Learning · SJSU Department of Applied Data Science

Aug 2026 to present

  • Support machine learning instruction alongside my research work.

Graduate Research Assistant · SJSU Department of Applied Data Science

Mar 2026 to present

  • Developed pre-training quality gates over Parquet metadata and H.264 video, validating episode and frame alignment, video decodability and resolution, and NaN-free action and proprioceptive tensors.
  • Engineered a reproducible PyTorch/CUDA training pipeline for 52M-parameter Action Chunking Transformer policies with configurable augmentation, Weights & Biases tracking, 5K-step checkpointing, and resume support, completing 30K- and 60K-step runs.
  • Built a resumable physical-robot evaluation pipeline for a 100-trial fixed and randomized cube-placement protocol with automatic return-to-start, stale-cache guards, placement-error measurement, and success/grasp/failure logging. Completed all 50 fixed-position rollouts at 92% success and 2.49 cm mean placement error.

Software Engineer · Sapaad

Jun 2023 to Oct 2024

  • Implemented a Delta Lake data quality framework with automated schema validation and anomaly detection for production ML.

Computer Vision Engineer Intern · Toothlens

Sep to Dec 2022

  • Developed computer vision pipelines for an orthodontic alignment assessment prototype, enabling automated detection of teeth positioning from smartphone images.
  • Optimized a PyTorch GAN pipeline by integrating OpenCV for geometric normalization, improving model recall by 9% and reducing image preprocessing failures.
  • Designed evaluation protocols to benchmark model performance across edge cases including partial occlusion and low resolution.

Research Assistant · Illinois Institute of Technology

Sep 2019 to Mar 2020

  • Automated tweet collection through the Tweepy API and generated FastText embeddings to identify water-related disaster events.
  • Trained an SVM classifier to 85% accuracy, adding GridSearchCV hyperparameter tuning for a further 7% gain.

Hiring for machine learning engineering?

I'm available from December 2026 and open to relocating. Email me and I'll reply within a day.

Hiring for something else?