Senior Artificial Intelligence Algorithms Engineer

$220,000 - $419,750/Yr

Nvidia - Santa Clara, CA

Full-time - Senior

Santa Clara, CA

Computer and Electronic Product Manufacturing

About the position

NVIDIA is seeking a Senior Artificial Intelligence Algorithms Engineer to design, develop, and optimize Artificial Intelligence solutions for a variety of real-world problems. This role requires a strong understanding of AI and Deep Learning, along with a deep algorithmic background and exposure to computer architecture and performance. The successful candidate will collaborate with internal partners, users, and members of the open-source community to analyze, define, and implement highly optimized AI algorithms. The responsibilities will include implementing new algorithms, performance and accuracy tuning, defining APIs, and analyzing functionality coverage to build larger, continuous toolsets and libraries. Excellent interpersonal skills and the ability to work in a multifaceted, product-centric environment are essential for success in this role. The position involves developing algorithms for AI, Deep Learning, data analytics, machine learning, or scientific computing. The engineer will tackle large-scale distributed systems capable of performing end-to-end AI training and inference deployment, which includes data fetching, pre-processing, orchestrating and running model training and tuning, and model serving. Additionally, the engineer will analyze, influence, and improve AI and Deep Learning libraries, frameworks, and APIs according to good engineering practices. Researching, prototyping, and developing effective tools and infrastructure pipelines will also be part of the job, along with publishing innovative results on platforms like GitHub and in scientific publications.

Responsibilities

• Develop algorithms for AI/DL, data analytics, machine learning, or scientific computing.
• Tackle large-scale distributed systems capable of performing end-to-end AI training and inference-deployment.
• Analyze, influence, and improve AI/DL libraries, frameworks, and APIs according to good engineering practices.
• Research, prototype, and develop effective tools and infrastructure pipelines.
• Publish innovative results on GitHub and in scientific publications.

Requirements

• A PhD or Master's Degree (or equivalent experience) and 5+ years of industry experience in Computer Science, AI, Applied Math, or related field.
• Strong Mathematical fundamentals and AI/DL algorithms skills or experience.
• Excellent programming, debugging, performance analysis, test design, and documentation skills.
• Experience with AI/DL Frameworks (e.g. PyTorch, JAX).
• Excellent C/C++ and Python programming skills.

Nice-to-haves

• Knowledge of GPU/CPU architecture and related numerical software.
• Prior experience with Generative AI techniques applied to Large Language Models and multimodal learning (Image, Video, Speech etc.).
• Exposure to large-scale AI training, understanding of compute system concepts (latency/throughput bottlenecks, pipelining, multiprocessing etc.) and related performance analysis and tuning.
• Hands-on experience with inference and deployment environments (e.g. TRT, ONNX, Triton).

Benefits

• Equity and benefits eligibility.