NVIDIA is looking to optimize Machine Learning training, inference, and NVIDIA AI Services by squeezing every last clock cycle out of their systems, and is seeking engineers to help achieve peak performance across the hardware/software stack.
Requirements
- Excellent Python programming skills, extensive knowledge of at least one ML Framework
- Algorithms and ML fundamentals
- Experience with Git or other source-control systems
- 2-4+ years of experience in ML model implementation and SW Development
- BS degree in Computer Science, Computer Architecture, Mathematics, Physics or related technical field or equivalent experience
Responsibilities
- Perform Exploratory Data Analysis (EDA) and document recommendations
- Experience with multiple ML frameworks (PyT, Jupyter Notebooks, Pandas and others)
- Able to performance inference and fine tuning on existing pre-trained models
- Basic knowledge of ML algorithms and evaluation techniques such as transformers, diffusion, LSTMs, SHAP
- Capability to script and run workloads on standard workload management platforms e.g. MLFlow, SLURM
- Collaborate with researchers and engineers across NVIDIA providing support on improving the design, usability and performance of models
Other
- BS degree in Computer Science, Computer Architecture, Mathematics, Physics or related technical field or equivalent experience
- 2-4+ years of experience in ML model implementation and SW Development
- Strong problem solving and analytical skills
- Published papers, patents or articles
- Ability to work in a diverse, supportive environment
