Software Engineer, PyTorch - Deep Learning
NVIDIA
Engineering & Technology
Job Summary
NVIDIA is hiring Deep Learning Compiler Engineers to design and build tools used by AI engineers across the world to design, develop, and deploy AI applications scalable across thousands of GPUs. Our team is responsible for the continual delivery of outstanding experience on NVIDIA's hardware with PyTorch. Apply for an opportunity to collaborate with many multi-disciplinary engineering teams within NVIDIA and internationally in the PyTorch open-source community to deliver our customers the best of NVIDIA software.
- Minimum Qualification: MBA / MSc
- Experience Level: Senior level
- Experience Length: 5 years
Job Description/Requirements
What You Will Be Doing:
- Conducting in-depth performance analysis of deep learning workloads, identifying bottlenecks, functional errors, and system inefficiencies.
- Correlating these performance issues with compiler bugs or missed optimization opportunities and developing strategies to address them.
- Collaborating with our team to extend existing program transforms or craft new ones based on the recommendations from performance analyses.
- Staying up-to-date with the latest advancements in deep learning compilers, and proposing innovative solutions to improve the efficiency of deep learning frameworks.
- Rigorously testing and validating compiler optimizations to ensure the highest quality and performance of model training.
What we need to see:
- Bachelors, Masters, or Ph.D. in Computer Science or a related technical field (or equivalent experience).
- Proficiency in Python.
- Experience using machine learning frameworks such as PyTorch or JAX.
- Some knowledge of compiler concepts such as abstract interpretation, code representations (e.g. SSA form, AST), code generation, and program transformations.
- Demonstrated experience developing large software projects.
Important Safety Tips
- Do not make any payment without confirming with the Jobberman Customer Support Team.
- If you think this advert is not genuine, please report it via the Report Job link below.