Lead Forward Deployed Software Engineer

Advanced Micro DevicesSanta Clara, CA
84dHybrid

About The Position

As a Forward Deployed Software Engineer, you will work closely with our most strategic partners as a hands-on technical expert. You are responsible for turning AMD's cutting-edge AI technology into tangible business value. This role is a unique blend of customer relationship skills and elite software engineer; you will work side-by-side with clients to help them prove out and ultimately deploy AI solutions on AMD GPUs. You will be the trusted technical advisor and hands-on developer who makes it happen.

Requirements

  • Strong programming skills in C/C++ and Python.
  • Experience with GPU kernel programming using CUDA, HIP or OpenCL.
  • Experience with collective communication and attentions.
  • Strong performance analysis and optimization skills for both CPU and GPU.
  • Expertise with modern AI/ML frameworks (e.g., PyTorch, TensorFlow, JAX).
  • Experience with distributed training and inference frameworks.
  • A track record of direct Client Engagement with experience working with customers to solve ambiguous technical problems.
  • Experience with containerization and orchestration technologies like Singularity, Docker, and/or Kubernetes.
  • Experience with open-source software development including collaboration with community maintainers and submitting contributions.
  • Experience with software engineering methodologies such as Agile, Scrum, Kanban.
  • Excellent analytical and problem-solving skills.
  • Ability to work independently and as part of a team.
  • BS/MS/PhD with related experience in Computer Science or related equivalent.

Nice To Haves

  • Experience in compiler, ISA.
  • Experience shipping software in an end-customer production environment.
  • Experience in implementing and optimizing collective communication primitives on GPU accelerators (NCCL/RCCL).
  • Experience in implementing and optimizing attention and KV-cache primitives on GPU accelerators.
  • Experience in all phases of software development, from requirement gathering, analysis, design, development, testing to final release.
  • Experience with test-driven development (TDD), CI/CD.
  • Experience providing clear and timely communication related to status and other key aspects of the project to the executive leadership team.

Responsibilities

  • Work closely with strategic customers to understand their requirements challenges and identify opportunities for AMD hardware and software to provide value.
  • Close gaps in the AMD software stack needed to support customer solutions.
  • Work hands-on as a technical expert and creative problem-solver, developing side-by-side with customers to drive projects from proof of concept to production.
  • Act as the voice of the customer, translating their needs and your on-the-ground insights into actionable feedback that shapes the future of AMD's AI roadmap.
  • Thrive in unfamiliar territory with a high degree of autonomy, finding novel ways to apply AMD technology to real-world problems.

Stand Out From the Crowd

Upload your resume and get instant feedback on how well it matches this job.

Upload and Match Resume

What This Job Offers

Career Level

Senior

Industry

Computer and Electronic Product Manufacturing

Education Level

Master's degree

Number of Employees

5,001-10,000 employees

© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service