Director – Research Engineering
Bristol Myers Squibb
AngularAWSCloudPythonReactSQL
Senior Research Engineer
NVIDIA
full-time
Posted on:
Location Type: Hybrid
Location: Santa Clara • California • 🇺🇸 United States
Visit company websiteSalary
💰 $160,000 - $299,000 per year
Job Level
Senior
Tech Stack
Open SourcePythonPyTorch
About the role
- Work with applied researchers to design, implement and test next generation of RL and pos-training algorithms
- Contribute and advance open source by developing NeMo-RL , Megatron Core, and NeMo Framework and yet to be announced software
- You will be engaged as part of one team during Nemotron models post-training
- Solve large-scale, end-to-end AI training and inference challenges, spanning the full model lifecycle from initial orchestration, data pre-processing, running of model training and tuning, to model deployment.
- Work at the intersection of computer-architecture, libraries, frameworks, AI applications and the entire software stack.
- Performance tuning and optimizations, model training with mixed precision recipes on next-gen NVIDIA GPU architectures.
- Publish and present your results at academic and industry conferences
Requirements
- BS, MS or PhD in Computer Science, AI, Applied Math, or related fields or equivalent experience
- 3+ years of proven experience in machine learning, systems, distributed computing, or large-scale model training.
- Experience with AI Frameworks such as Pytorch or JAX
- Experience with at least one inference and deployment environments such as vLLM, SGLang or TRT-LLM
- Proficient in Python programming, software design, debugging, performance analysis, test design and documentation.
- Strong understanding of AI/Deep-Learning fundamentals and their practical applications.
Benefits
- equity
- benefits 📊 Resume Score Upload your resume to see if it passes auto-rejection tools used by recruiters
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
machine learningdistributed computinglarge-scale model trainingPython programmingperformance tuningmodel trainingmixed precision recipesAI/Deep-Learning fundamentalstest designdebugging
Certifications
BS in Computer ScienceMS in Computer SciencePhD in Computer ScienceBS in AIMS in AIPhD in AIBS in Applied MathMS in Applied MathPhD in Applied Math