Sr AI Software Development Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Advanced Microdevices Pvt. Ltd. (India) · 17 hours ago

Sr AI Software Development Engineer

Advanced Micro Devices, Inc is focused on building innovative products that enhance next-generation computing experiences. They are seeking a highly motivated AI Software Engineer to join their team, responsible for enabling software for datacenters and optimizing performance for GPU computational accelerators.

BiopharmaBiotechnologyIndustrialManufacturing

Responsibilities

Would you like to be part of a world class team enabling software for world class datacenters and the mightiest supercomputers?
AMD is searching for talented and highly motivated AI Software Engineers to join our team of developers pushing the boundaries of efficiency and performance to enable and optimize the software ecosystem for the next generation of GPU computational accelerators
Our team has an unparalleled perspective of the AI landscape
We work with the industry’s most sophisticated clients to help them leverage the latest hardware capabilities for their AI use cases
As part of our team, you will be among the first in the world to combine the newest hardware with the industry’s latest applications, libraries, frameworks, and SDKs to push the limits of innovation and solve the world’s most complex challenges
You will work with a team of Software Engineers to enable DL models, libraries, and applications for Instinct GPUs in both on-prem and Cloud environments

Qualification

C++PythonDeep Learning frameworksGPU performance optimizationOpen-source software developmentPerformance analysisAnalytical skillsWillingness to learnPublications in ML conferencesDockerKubernetesProblem-solving skillsTeam collaborationCommunication skills

Required

Strong programming skills in C++ and Python
Strong development experience is at least one major DL framework such as Pytorch or Tensorflow in inference, fine tuning and/or training
MS or PhD in Computer Science or Computer Engineering or related equivalent
Experience developing software and system-level performance optimizations with a solid architecture understanding in GPUs a plus
Experience with open-source software development including collaboration with community maintainers and submitting contributions is a plus
Publications in reputed peer-reviewed ML conferences/journals a plus
Excellent analytical and problem-solving skills root-causing/addressing performance issues
Ability to work independently and as part of a team
Willingness to learn skills, tools, and methods to advance the quality, consistency, and timeliness of AMD software products

Preferred

Expertise in profiling tools across the AI SW Stack (Torchprofiler, RocM profiler, Vtune, Nsight)
Experience in implementing and optimizing parallel methods on GPU accelerators (NCCL/RCCL, OpenMP, MPI)
Performance analysis skills for both CPU and GPU
Experience with Singularity, Docker, and/or Kubernetes
Experience providing clear and timely communication related to status and other key aspects of the project to leadership team

Benefits

AMD benefits at a glance.

Company

Advanced Microdevices Pvt. Ltd. (India)

twittertwittertwitter
company-logo
Advanced Microdevices (mdi) is a leader in innovative membrane technologies.

Funding

Current Stage
Late Stage

Leadership Team

leader-logo
Nalini Kant Gupta
Founder & Managing Director
Company data provided by crunchbase