LLM Training Performance Engineer; GPU Optimization
Listed on 2026-11-05
-
Software Development
AI Engineer (Applied/Software), Machine Learning/ ML Engineer, AI Reliability/ Performance Engineer
Location: Town of Montana
NVIDIA Corporation seeks an experienced High-Performance LLM Training Engineer to optimize its AI training workloads across frameworks like PyTorch and JAX for thousands of GPUs. You will profile, tune, and implement production-ready software across the deep learning stack and help shape future GPU architectures.
Required are advanced degrees in CS/EE/CE and a strong background in DL, model training optimizations, and GPU architecture. Proficiency in C, Python, and CUDA is essential.
Are you ready to take on the LLM Training Performance Engineer (GPU Optimization) role at NVIDIA Corporation?
Join NVIDIA Corporation and contribute to our ongoing work.
Take a moment to read everything above and see whether this role is right for you.
This posting is for the LLM Training Performance Engineer (GPU Optimization) role at NVIDIA Corporation, based in Town of Montana, WI, United States.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).