Senior LLM Train Framework Engineer
NVIDIA(4 months ago)
About this role
An LLM Train Framework Engineer on NVIDIA's Megatron Core team contributes to an open-source, scalable, cloud-native framework used by researchers and developers for large language and multimodal foundation model development. The role supports the full model lifecycle across pretraining, alignment, customization, evaluation, deployment, and tooling to improve model performance and developer experience.
Required Skills
- Python
- PyTorch
- JAX
- CUDA
- Distributed Training
- Model Parallelism
- Mixed Precision
- Performance Tuning
- Debugging
- Data Preprocessing
+5 more
Qualifications
- MS in Computer Science
- PhD in Computer Science
- MS in Applied Math
- PhD in Applied Math
About NVIDIA
nvidia.comNVIDIA invents the GPU and drives advances in AI, HPC, gaming, creative design, autonomous vehicles, and robotics.
View more jobs at NVIDIA →Apply instantly with AI
Let ApplyBlast auto-apply to jobs like this for you. Save hours on applications and land your dream job faster.
More jobs at NVIDIA
Similar Jobs
AI Frameworks Software Engineer - vLLM
Intel(3 months ago)
Research Internship- Multimodal LLM (Speech/Music/Audio/Vision/Language)
Tencent(2 months ago)
Generative AI - ML System Engineering
Meshy(1 year ago)
ML Engineer, Large Language Models (LLM Training & Inference Optimization)
Nebius(10 months ago)
Research Engineer Intern
Hiverge(2 months ago)
AI Inference Engineer - Speech
Zoom(1 month ago)