G

Gautam Jain

AI Researcher

United States6 yrs 1 mo experience
Highly Stable

Key Highlights

  • Expert in developing optimized deep learning kernels.
  • Led significant projects at Nvidia and Citadel Securities.
  • Strong academic background in Computational Data Science.
Stackforce AI infers this person is a Deep Learning Engineer with expertise in GPU programming and artificial intelligence.

Contact

Skills

Core Skills

Deep LearningGpu Programming

Other Skills

Optimized KernelsC++PythonMachine LearningAlgorithmsData ScienceProbabilityNatural Language Processing (NLP)Data MiningNeural Networks

About

Senior Research Engineer at Citadel Securities Senior deep learning library engineer at Nvidia cuDNN and team lead for the attention project in cuDNN developing optimized state of the art flash attention kernels across GPU generations. Graduate Student at Carnegie Mellon University with a Masters in Computational Data Science. Capstone Project with Prof Andy Pavlo in query optimizer, particulary related to adding CTE Support to CMU-DB. Mathematics and Computing undergrad at IIT Delhi with one year experience as a Quantitative Researcher at WorldQuant India.

Experience

6 yrs 1 mo
Total Experience
2 yrs 5 mos
Average Tenure
1 yr 2 mos
Current Experience

Citadel securities

Senior Research Engineer

Feb 2025Present · 1 yr 2 mos · Miami, Florida, United States

Nvidia

3 roles

Senior Deep Learning Engineer

Promoted

Jul 2022Jan 2025 · 2 yrs 6 mos · Santa Clara, California, United States

Deep LearningGPU ProgrammingOptimized Kernels

Deep Learning Library Software Engineer

Feb 2021Jul 2022 · 1 yr 5 mos · Santa Clara, California, United States

Deep Learning Library SWE Intern

May 2020Aug 2020 · 3 mos · Santa Clara County, California, United States

Worldquant

Quantitative Researcher

Jul 2018Jul 2019 · 1 yr · Mumbai Area, India

Itc limited

Intern

May 2017Jul 2017 · 2 mos

Education

Carnegie Mellon University

Masters — Computational Data Science

Jan 2019Jan 2020

Indian Institute of Technology, Delhi

Bachelor’s Degree — Mathematics and Computer Science

Jan 2014Jan 2018

Stackforce found 100+ more professionals with Deep Learning & Gpu Programming

Explore similar profiles based on matching skills and experience