Senior Machine Learning Performance Engineer – Physics
NVIDIA
full-time
Posted on:
Location Type: Remote
Location: California • United States
Visit company websiteExplore more
Salary
💰 $152,000 - $241,500 per year
Job Level
Tech Stack
About the role
- Work with some of the brightest minds in a leading AI company to develop a leading machine learning framework, NVIDIA PhysicsNeMo, for our academic and industrial partners to construct digital twins and machine learning simulation surrogates for real world science and engineering problems
- Work with internal project teams to validate applications built using the framework on Nvidia’s products
- Characterize and optimize scientific AI workloads on the latest NVIDIA platforms
- Stay up to date with the latest research and innovations in deep learning techniques, implement and experiment with new ideas to develop and enhance NVIDIA's deep learning technologies with focus on simulations
Requirements
- BS or MS degree (PhD preferred) in computer science, mathematics, computational science/engineering, or related technical field or equivalent experience
- 5+ yrs of relevant experience
- Strong Python programming skills
- Familiarity with containers, numeric libraries, modular software design, distributed computing, and/or high-performance storage systems
- Good knowledge of state-of-the-art DNN architectures and machine learning techniques and algorithms (graph networks, diffusion models, reinforcement learning etc.) with experience in developing or using major deep learning frameworks (PyTorch, Tensorflow, JAX etc.)
- Experience with solving and using machine learning for real world problems involving scientific/engineering simulations
- Experience with scientific visualization is a big plus
- Experience with performance measurement tools, such as NVIDIA Nsight Systems
- Strong analytical skills with bias for action
- Good time-management and organization skills to thrive in a fast paced, dynamic environment
- Solid written and oral communications skills
- Good teamwork and interpersonal skills
Benefits
- health insurance
- retirement plans
- paid time off
- flexible work arrangements
- professional development
- bonuses
- stock options
- equipment allowances
- wellness programs
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard Skills & Tools
Pythondeep learningDNN architecturesmachine learning techniquesgraph networksdiffusion modelsreinforcement learningPyTorchTensorFlowJAX
Soft Skills
analytical skillstime-managementorganization skillscommunication skillsteamworkinterpersonal skills
Certifications
BS degreeMS degreePhD preferred