Senior ML Framework Performance Engineer - AI for Science at Scale
NVIDIA(10 days ago)
About this role
NVIDIA is seeking a Senior Machine Learning Performance Engineer to advance scientific machine learning frameworks, particularly in atomistic modeling. The role involves evaluating ML model performance, identifying bottlenecks, and collaborating with research teams to improve AI for Science applications. The position offers an opportunity to work on cutting-edge technologies in AI and scientific computing.
Required Skills
- PyTorch
- JAX
- Warp
- Distributed Learning
- Model Parallelism
- CUDA
- Triton
- Computational Biology
- Scientific Computing
- Parallel Computing
Qualifications
- Advanced degree in a quantitative field
- 5+ years of experience
- Performance Engineering
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
ML Engineer, Biological Analysis & Simulation
Mithrl(1 month ago)
Postdoctoral Appointee - Exascale Molecular Simulations
UChicago Argonne(1 month ago)
Postdoctoral Research Associate
Brookhaven National Laboratory(2 days ago)
Senior Machine Learning Engineer (Large Systems)
Graphcore(30 days ago)
Senior Machine Learning Engineer (Large Systems)
Graphcore(30 days ago)
Machine Learning Researcher
Astera Institute(3 days ago)