AI Networking Software Developer

Yesterday

Apply Now

Receive Emails with Similar Jobs

NVIDIA

WebsiteLinkedInAll Job Openings

GPU-accelerated computing • artificial intelligence • deep learning • virtual reality • gaming

10,000+

Description

• Design and Implement features for AI communication middleware and networking libraries to improve performance, resiliency and scalability. • Develop features for libraries such as NCCL, Open MPI, UCX, UCC, NVSHMEM, and VERBS • Develop features for Deep Learning frameworks such as PyTorch and TensorFlow • Creating proofs-of-concept to evaluate and motivate extensions in AI Frameworks (PyTorch/NEMO), HPC programming models (MPI, OpenSHMEM, PGAS), new runtime designs, and new network hardware features.

Requirements

• A Bachelors. or Masters, in computer science, computer engineering, or a closely related field or equivalent experience. • 5+ years of experience in parallel programming models, and/or network architecture. • Background in algorithm design, system programming, and computer architecture • Strong programming and software development skills • Ability and flexibility to work and communicate effectively in a multi-national, multi-time-zone corporate environment

Benefits

• Competitive salaries • Generous benefits package • Eligibility for equity