NVIDIA is a leading technology company known for groundbreaking developments in AI and High-Performance Computing
Job Summary
NVIDIA is a leading technology company known for groundbreaking developments in AI and High-Performance Computing.
The role involves acting as a full-stack AI problem solver, designing ablation studies, writing high-performance code, and upstreaming learnings into core libraries.
Candidates will lead complex training experiments, design custom benchmarks, and drive inference efficiency to deliver fast, deployable AI models.
Matching Summary
NVIDIA is a leading technology company known for groundbreaking developments in AI and High-Performance Computing.
Skills & Requirements
Must-have
LLM/Deep Learning lifecycle expertise
Expert-level Python and PyTorch
Experience with Megatron-LM and NeMo frameworks
Model training and evaluation
Inference optimization and compression techniques
Nice-to-have
Open source contributions to LLM projects
High-energy and intrinsically motivated
Deep understanding of Transformers and scaling laws
Familiarity with NVIDIA AI stack and NIM ecosystem
Key Requirements
Masters or PhD in Computer Science or related field