AI Software Architect, AI Software Solutions

AMD - Santa Clara, CA

Full-time - Mid Level

Santa Clara, CA

Computer and Electronic Product Manufacturing

About the position

The AI Software Architect position at AMD focuses on enabling software for advanced data centers and supercomputers, particularly through the development of AI-based solutions. The role involves working with a team to optimize software ecosystems for next-generation GPU computational accelerators, leveraging the latest hardware capabilities to address complex challenges in AI and computing.

Responsibilities

• Enable deep learning models, libraries, and applications for Instinct GPUs in both on-prem and cloud environments.
• Analyze and optimize the performance of AI software, addressing hardware bottlenecks to maximize efficiency.
• Collaborate with a team of software engineers to push the boundaries of performance and innovation in AI software solutions.
• Work with sophisticated clients to leverage hardware capabilities for their AI use cases.
• Contribute to open-source software development and collaborate with community maintainers.

Requirements

• Minimum 4 years of experience in AI software engineering or related fields.
• Strong programming skills in C++ and Python.
• Development experience with at least one major deep learning framework such as PyTorch or TensorFlow.
• Experience in software and system-level performance optimizations with a solid understanding of GPU architecture.
• Excellent analytical and problem-solving skills, particularly in addressing performance issues.

Nice-to-haves

• Expertise in profiling tools across the AI software stack (e.g., Torchprofiler, RocM profiler, Vtune, Nsight).
• Experience in implementing and optimizing parallel methods on GPU accelerators (NCCL/RCCL, OpenMP, MPI).
• Familiarity with containerization technologies such as Singularity, Docker, and Kubernetes.
• Publications in reputed peer-reviewed ML conferences or journals.

Benefits

• Base pay competitive with industry standards.
• Annual bonus or sales incentive eligibility based on role.
• Opportunity to own shares of AMD stock through the Employee Stock Purchase Plan with discounts.