Naveenraj Kamalakannan
I am an AI/ML Associate at J.P. Morgan (Asset & Wealth Management) in New York, where I build agentic AI applications for investment workflows and drive performance engineering of the LLM serving stack - latency/throughput profiling, inference optimization, and capacity planning for production agentic workloads.
I completed my M.S. in Computer Engineering at New York University (May 2026), where I was a Research Assistant at the NYU Center for Data Science and NYU Langone, working on sub-second action detection with vision-language models for stroke rehabilitation, advised by Prof. Carlos Fernandez-Granda and Prof. Heidi Schambra. I earned my Bachelor's in Electronics Engineering from Vellore Institute of Technology.
I work on LLM inference & training systems - quantization, kernel-level performance, and serving - and contribute across the stack: vLLM, DeepSpeed, SGLang, TRT-LLM, NVIDIA's OSS ecosystem, and other cracked open-source initiatives.
Email / GitHub / Google Scholar / LinkedIn / CV
Totum
Totum is my research project on rebuilding LLM inference from first principles: what does a serving engine look like when phase-aware kernel dispatch, quantization-native execution, and heterogeneous hardware are the starting point instead of afterthoughts?
The engine comes first. Around it, a platform: fine-tuning, hosting, bring-your-own-cloud.
Cooking right now 🍳
Open Source Contributions
-
DeepSpeed
DeepCompile compiler pass for AutoTP New collective primitives + FX-graph rewrites that defer tensor-parallel communication into the compiled graph (part 1 of 2 merged).
-
TensorRT-LLM
Tree-of-Thought & MCTS controllers in AutoDeploy ToT and MCTS integration in the AutoDeploy scaffolding framework for search-based inference strategies.
-
vLLM
Separate MLAAttention from Attention Decoupled MLA prefill/decode paths from a unified custom op, enabling torch.compile fusion and piecewise CUDA-graph capture.
-
ArcticInference
FlashInfer backend for SwiftKV FlashInfer support in Snowflake's SwiftKV path for high-throughput KV-cache-aware decoding.
-
DeepSpeed
ZeRO-3 CPU-offload gradient clipping fix Global gradient norms now reflect clipped gradients under CPU offload; follow-up DeepNVMe Zip-serialization work with PyTorch Core.
Research & Publications
The Potential and Limitations of Vision-Language Models for Human Motion Understanding
PLOS Digital Health · 2025
Benchmarked SOTA VLMs (InternVL3, NVILA, LLaVa-OneVision) and engineered a pose-refined prompting pipeline that integrates YOLOv11 pose tracks with VLM context to extract sub-second motion primitives, achieving ~67.75 Edit Score (ES) on structured upper-limb rehab tasks.
Exponential Pixelating Integral Transform with Dual Fractal Features for Enhanced Chest X-Ray Abnormality Detection
Computers in Biology and Medicine, Volume 182 · 2024
Built a chest X-ray abnormality detection model using Exponential Pixelating Integral transforms and fractal features. Implemented Multivariate Adaptive Regression Splines (MARS) ensemble, achieving 99.63% accuracy and F1 scores up to 98.10%.
A Novel Approach for the Early Detection of Parkinson's Disease Using EEG Signal
IJEET 12.5 (2021): 80-95 · 2021
Led a team to develop a Deep Learning model analyzing EEG signals, achieving 93.3% accuracy in detecting early-stage Parkinson’s disease. Attained an F1 score of 93.48% and presented findings at the University of Tubingen Symposium.
Experience
AI/ML Associate - J.P. Morgan
J.P. Morgan Asset & Wealth Management, New York · July 2026 - Present
Building agentic AI applications for investment workflows, taking multi-agent systems from prototype to scalable production deployments on the firm’s agentic platform. Driving performance engineering of the LLM serving stack: latency/throughput profiling, inference optimization, and capacity planning for production agentic workloads.
AI & Data Science Associate Intern - J.P. Morgan
New York · June 2025 - August 2025
Architected and enhanced Asset and Wealth Management’s agentic platform, improving the relevance and naturalness of AI-generated content through a multi-stage retrieval pipeline. Orchestrated multiple AI agents with advanced reasoning capabilities to significantly enhance existing product functionality. Applied expertise in OpenSearch, RAG Systems, LangChain, KV Cache Optimization, and MCP Server to develop robust and performant AI solutions for financial applications.
Research Assistant - NYU Center for Data Science & NYU Langone
NYU Center for Data Science & NYU Langone, New York · February 2025 - May 2026
Working under the supervision of Prof. Carlos Fernandez-Granda, Prof. Heidi Schambra and Victor Li. Studied whether current VLMs can recover clinically meaningful upper-limb motion primitives from rehab videos. Benchmarked SOTA VLMs (InternVL3, NVILA, LLaVa-OneVision) and engineered a pose-refined prompting pipeline that integrates YOLOv11 pose tracks with VLM context to extract sub-second motion primitives, achieving ~67.75 Edit Score (ES) on structured upper-limb rehab tasks and exposing systematic failure modes on subtle hand-object interactions.
Control Engineer - Zeeco Middle East
Zeeco Middle East, Dammam, Saudi Arabia · July 2022 - August 2023 (Full-time) | Summer 2019 (Internship)
ML-based anomaly detection for industrial circuit designs and predictive control (PID + neural networks) for pressure/temperature/flow optimization on plant equipment.
Data Engineer Intern - Bajaj Finserv
Bajaj Finserv, Pune, India · January 2022 - June 2022
Streamlined Azure data-migration pipelines (EDW → Cosmos DB), cutting migration time by 34%, and built an ML model that reduced Azure EDW resource consumption by ~33%.
Projects
Starbots.AI Automated Cafeteria System
Robotics Project - The Construct Bootcamp · 2024
Built an autonomous mobile robot using ROS2, RRT* path planning, and OMPL for navigation. Integrated CNNs for object detection, SLAM for mapping, and robotic manipulation for food handling. Developed with Python, C++, and Gazebo simulation as part of The Construct Robotics Bootcamp.
Adaptive Monte-Carlo Localization Warehouse Robot
Robotics Project - The Construct Bootcamp · 2024
Built an autonomous warehouse robot using Adaptive Monte-Carlo Localization (AMCL) and Cartographer SLAM for precise positioning. Implemented Nav2 navigation stack with costmap generation, path planning, and obstacle avoidance. Developed with ROS2, Python, and Gazebo simulation for the RB1 robot platform.
Early Detection of Sepsis - National Hackathon Winner
VIT National Hackathon · First Place - Design Category · 2020
Led a cross-functional team to develop a sepsis onset detection model using critical biomarkers (PCT and MDW), securing first place in the Design Category and winning a $2,000 grant from the VIT Incubator. The project focused on early detection of sepsis, a critical medical condition requiring rapid intervention.