Siddharth Mehta

Siddharth Mehta

Data Scientist III @ Walmart | Applied AI Engineer
I build applied AI systems across agentic workflows, large-scale ML platforms, and research-driven optimization problems.
Buffalo, New York • from 🇮🇳
  • Applied AI
  • Agentic Systems
  • ML Platforms
  • Research Engineering
  • MLOps
  • Full-Stack Systems

Building customer-facing agent capabilities at Walmart using MCP-style skills and tool integrations.

Delivered production ML and LLM systems across Walmart, ZS, and research environments with measurable business impact.

Published and presented research in imaging, optimization, and biomedical signal modeling.

I work at the intersection of applied AI, machine learning systems, and research engineering. My background spans production ML, agentic workflows, experimentation platforms, and optimization-heavy research problems. Explore my projects to see how I build, or read my articles for a deeper view into how I think about systems, modeling, and engineering tradeoffs.

Project Highlights

View More

Selected Experience

Walmart Global Tech

Walmart Global Tech

Data Scientist III

Feb 2026 - Present

Building applied AI capabilities for customer assistance by enabling Super Agents with MCP-style servers, skills, and tool integrations that support task execution and richer agent workflows.

Visit
Graduated from the University at Buffalo — M.S. in Computer Science (Jan 2026)
University at Buffalo

University at Buffalo

Research Assistant

Aug 2025 - Dec 2025

Developed quantitative sampling-coverage metrics and optimized physics-based imaging systems through 3D simulation, vectorized PyTorch ray-tracing (85% speedup), and Bayesian optimization for large configuration search, enabling scalable experimental workflows and contributing to publications at SPIE Imaging, UNYAPM, and IEEE TMI.

Visit
Walmart Global Tech

Walmart Global Tech

Data Science Intern

Jun 2025 - Aug 2025

Built and deployed large-scale ML/LLM systems for content understanding, distribution-shift monitoring, automated evaluation, and personalized feature ranking, extracting high-signal insights from thousands of artifacts while reducing manual effort by 95% and delivering autoscaled real-time performance on GCP.

Visit
ZS Associates

ZS Associates

Data Science Associate

Jun 2022 - Jul 2024

Developed large-scale PySpark/Hadoop pipelines and XGBoost-based prioritization models to power product analytics and next-best-engagement decisions, reducing rep workload by 20%, improving targeting accuracy, and driving $100K+ incremental revenue through data-driven customer-level actions.

Visit
Lakebrains Technologies

Lakebrains Technologies

Machine Learning Intern

May 2020 - Jul 2020

Engineered real-time edge-compute systems combining face recognition with motion-detection–based activity recognition, optimized and deployed on Raspberry Pi and Jetson Nano for efficient, low-latency performance.

Graduated from NITH — B.Tech in Computer Engineering (Class of 2022)
National Institute of Technology Hamirpur

National Institute of Technology Hamirpur

Research Assistant — Biometric Signal Analysis

Aug 2020 - Jun 2022

Conducted research on ECG and PPG signal processing for non-invasive blood pressure estimation, implementing hybrid neural network models and validating performance on the MIMIC-II dataset.

Visit

Leadership & Community

Additional roles that reflect how I contribute beyond core engineering work: organizing teams, representing peers, and building communities.

National Institute of Technology Hamirpur
National Institute of Technology Hamirpur
Training & Placement Representative
Aug 2021 - Jun 2022

Coordinated with companies and students to streamline the on-campus recruitment process, managed data for candidate shortlisting, and supported logistics for placement drives.

Visit
Rhythmeecz dance club - National Institute of Technology Hamirpur
Rhythmeecz dance club - National Institute of Technology Hamirpur
Dance Club Secretary
Aug 2021 - Jun 2022

Led the official dance club, organized cultural events and performances, managed team operations, and represented the institute in inter-college competitions.

Visit