AI Data Specialist
RWS Group
Senior HPC Performance Engineer, AI for Science at Scale
NVIDIA
full-time
Posted on:
Location Type: Remote
Location: Remote • California, Massachusetts, North Carolina, Oregon, Texas • 🇺🇸 United States
Visit company websiteSalary
💰 $184,000 - $356,500 per year
Job Level
Senior
Tech Stack
PythonPyTorchTensorflow
About the role
- Design and implement computationally performant features for large scale, CUDA-backed ML training frameworks, using low level acceleration and scaling strategies such as GPU porting, data structure innovations, distributed learning technologies
- Optimize computational performance of wide range of business-critical ML models via accelerated hardware and software stack, as well as algorithmic improvements
- Develop and maintain HPC software stack for generative machine learning models in digital biology and beyond
- Collaborate with multiple HPC, AI infrastructure, and research teams
- Develop tools to assist data processing, data quality control, algorithm development, and algorithm testing
- Drive the testing and maintenance of the algorithms and software modules
Requirements
- Advanced degree in a quantitative field such as Computer Science, Computational Biophysics, Computational Chemistry, Physics, Mathematics, or equivalent experience
- 8+ years of relevant experience
- Consistent track record in performance engineering as well as software design, building and packaging and launching software products
- Deep understanding of parallel programming in C++ and Python; ideally CUDA and OAI Triton programming experience
- Fluent in modern machine learning frameworks such as PyTorch, TensorFlow, JAX, Warp
- Experience with HPC solutions to research problems, ideally for biology, chemistry or material science applications
- Recognized for technical leadership contributions, capable of self-direction, and ability to learn from and teach others
- Strong communication skills, organized and self-motivated, excellent teammate
- (Ways to stand out) Contributor to major scientific AI for Science codebase
- (Ways to stand out) Familiarity with pioneering language and geometric models used in AI for Science applications in biology, chemistry, material science
Benefits
- With highly competitive salaries and a comprehensive benefits package
- You will also be eligible for equity and benefits
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
CUDAC++Pythonperformance engineeringparallel programmingPyTorchTensorFlowJAXHPCalgorithm development
Soft skills
technical leadershipself-directioncommunicationorganizationself-motivationteamworkteaching
