Principal Artificial Intelligence Algorithms Engineer

Invidia

Multiple Locations
Base: 272,000 usd - 431,250 usd; bonus/equity: eli...
Ai frameworks development
Distributed training algorithms
Model parallelism techniques
NVIDIA is seeking engineers to expand capabilities of Megatron Core and NeMo Frameworks for large language and multimodal model training and optimization

Job Summary

  • NVIDIA is seeking engineers to expand capabilities of Megatron Core and NeMo Frameworks for large language and multimodal model training and optimization.
  • The role involves developing distributed training algorithms, optimizing performance on next-gen NVIDIA GPU architectures, and contributing to open source AI tools.
  • NVIDIA offers a competitive base salary range, equity, benefits, and a diverse and inclusive work environment.

Matching Summary

NVIDIA is seeking engineers to expand capabilities of Megatron Core and NeMo Frameworks for large language and multimodal model training and optimization.

Salary

Base: 272,000 USD - 431,250 USD; Bonus/Equity: Eligible for equity; Benefits: Eligible for benefits

Skills & Requirements

Must-have

  • AI Frameworks development
  • Distributed training algorithms
  • Model parallelism techniques
  • Python programming proficiency
  • Performance tuning on NVIDIA GPUs
  • Open source contributions

Nice-to-have

  • Generative AI techniques experience
  • Knowledge of GPU/CPU architecture
  • Large-scale AI training expertise
  • Cross-team collaboration skills
  • Innovative AI library improvements

Key Requirements

  • MS, PhD or equivalent experience
  • 10+ years industry experience
  • Experience with AI frameworks like PyTorch or JAX
  • Proficient in software design and debugging
  • Strong AI/deep learning fundamentals

Work Rights

Not specified

Tailored Resume

Cover Letter