Akshay Subramaniam
Akshay Subramaniam is an AI researcher at NVIDIA whose work applies machine learning and GPU computing to scientific simulation and weather prediction. He earned a bachelor's degree in aerospace engineering at IIT Madras in 2012 and an MS in aeronautics and astronautics at Stanford in 2014 before completing a PhD focused on turbulent flows, computational physics, numerical algorithms, high-performance computing and deep learning. He coauthored NVIDIA's GPU-accelerated Wide & Deep recommender implementation, SimNet and ClimSim; ClimSim won a 2023 NeurIPS Outstanding Paper award. His later work added hydrostatic constraints and generative AI to weather forecasting; since March 2026, he is a Principal AI Research Engineer, DevTech, at NVIDIA.
Timeline
-
2016
Shock-induced turbulent mixing
At Stanford, he presents a numerical study of shock-induced turbulent mixing in an inclined-interface Richtmyer–Meshkov instability in 2016.
-
2017
Plastic flow at interfaces
At Stanford's Thermal and Fluid Sciences Affiliates Program, he presents research on shock-driven plastic flow at solid-solid interfaces in 2017.
-
2012–2018
Degrees before NVIDIA
He earns a bachelor's degree in aerospace engineering at IIT Madras in 2012 and an MS in aeronautics and astronautics at Stanford in 2014. By November 2018, he completes a Stanford PhD focused on turbulent flows, computational physics, numerical algorithms, high-performance computing and deep learning, then joins NVIDIA as a senior AI developer technology engineer.
-
2019
Wide & Deep on GPUs
In 2019, he coauthors NVIDIA's GPU-accelerated Wide & Deep recommender implementation, which reduces inference latency by up to 13x or improves throughput by up to 11x against CPU baselines.
-
2021
NVIDIA SimNet
In 2021, he coauthors NVIDIA SimNet, an AI-accelerated multi-physics simulation framework.
-
2023
ClimSim earns a NeurIPS award
In 2023, he coauthors ClimSim, a large multi-scale dataset for hybrid physics-ML climate emulation. NeurIPS gives the paper its Outstanding Paper award.
-
June 2025–November 2025
Weather prediction at NVIDIA
In June 2025, he coauthors a global machine-learning weather prediction study that imposes hydrostatic-balance constraints; its soft constraint improves the RMSE of many forecast fields, especially beyond 7–10 days. By November 2025, he is a senior AI developer technology engineer at NVIDIA, focused on AI-powered climate modeling and petascale data workflows. In November 2025, he coauthors NVIDIA's article 'Gen AI Super-Resolution Accelerates Weather Prediction with Scalable, Low-Compute Models.'
-
Since March 2026
Principal AI Research Engineer
Since March 2026, he is a Principal AI Research Engineer, DevTech, at NVIDIA. In June 2026, he coauthors STRATA, an autoregressive AI emulator for global storm-resolving atmospheric dynamics trained on 17 days of SCREAM physics-model output at 4.9-km resolution sampled every 10 minutes.
Career
- Since 2026NVIDIASenior AI Developer Technology Engineer
Education
- 2012IIT MadrasBachelor’s degree in aerospace engineering
- Stanford UniversityPhD
Sources and credits
Photo: We have no published photo of Akshay Subramaniam yet. Send us a free-licensed photo.
- 2016 Posters | Aeronautics and Astronautics
- Presentations - 2017 | Thermal & Fluid Sciences Affiliates Program
- A High Resolution Compact Scheme for Compressible Flows Involving Shocks and Turbulence
- Presenter – SC25 Schedule
- Find Akshay Subramaniam's email & phone number - Principal AI Research Engineer, DevTech at NVIDIA, Stanford | SignalHire
- Accelerating Wide & Deep Recommender Inference on GPUs | NVIDIA Technical Blog
- NVIDIA SimNet: An AI-accelerated multi-physics simulation framework | NVIDIA Learning and Perception Research
- ClimSim: A large multi-scale dataset for hybrid physics-ML climate emulation | NVIDIA Climate
- NeurIPS 2023 Awards
- Imposing the Fundamental Dynamical Constraint of Hydrostatic Balance to Improve Global ML Weather Prediction
- Gen AI Super-Resolution Accelerates Weather Prediction with Scalable, Low-Compute Models | NVIDIA Technical Blog
- Scaling Storm-Resolving Atmospheric AI Simulation to the Entire Planet


