AI Technical Specialist

WHAT YOU DO AT AMD CHANGES EVERYTHING

We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world’s most important challenges. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives. 

AMD together we advance_

The Role

AMD is seeking an experienced machine learning developer to engage our customers, partners, and business teams and support solutions that accelerate adoption of AMD Pervasive AI products.  Successful candidates will be responsible as a trusted advisor for the analysis and support of customer machine learning workloads and proof-points that demonstrate optimized implementation within specific application segments of AMD’s AI portfolio of products.  In this capacity, you will help support the business for the expansion of AMD’s Pervasive AI strategy. 

You will engage and provide product and application domain expertise and support directly with technical, business and sales teams working with customer applications. The other responsibilities will include implementation of machine learning models, applications and demos on AMD products and provide domain expertise and advice on mapping and implementation of machine learning applications across the AMD AI product portfolio.

As a member of the Developer Acceleration Team within AMD’s AI Product Management organization, you will have opportunity to work closely with internal stakeholders across the company as well as engage directly with external customers.  Close interaction with Engineering, Applications, and Business Unit stakeholders is essential.

The Person

The type of person who will be successful in this role is highly collaborative, and willing to trailblaze and champion new ideas. You will exhibit excellent written and verbal communication skills and you will be highly analytical.  Applicants should possess a MS/PhD in EE or equivalent. The position requires comprehensive knowledge of machine learning architectures, models, and developer tools. Prior work experience in accelerated machine learning is required.

Key Responsibilities:

• Support and deliver technical expertise to support customer success in deployments of machine learning solutions on AMD products
• Providing domain expertise in AMD silicon, IP, and software including compilers, libraries, and profiling tools
• Provide domain expertise in foundational machine learning architectures, tools, and developer flows
• Implementation and training of machine learning models in common frameworks including PyTorch, Keras/Tensorflow, ONNX
• Experience with deployed production runtime serving of machine learning applications
• Experience publishing to open source, including github repositories and industry collaborations
• Experience with HIP, OpenAI Triton, Cuda or other kernel development programming flows
• Developing in Python, C/C , within Jupyter notebooks
• End to end performance testing and optimization of machine learning workloads
• Application profiling with CPU, GPU, NPU and accelerator tools (VTune, NVprof, etc)
• Integration and development of demonstration/application software

 Preferred Experience in following skills

• Experience with foundational machine learning model architecture development or implementation including LLMs, Embeddings, RAG, CNNs, Stable Diffusion, ASR, and NLP pipelines
• Domain expertise in one or more of the following application domains such as Computer Vision (CV), AI Assistants, Robotics, Smart City, Industrial Automation, Autonomous Driving, Natural Language Understanding (NLU/NLP), Graph Analytics, Recommendation Systems, Anomaly Detection, and High Performance Computing (HPC)
• Perl/Python/C scripting for test automation
• Effective communication and problem-solving skills
• Motivating leader with good interpersonal skills

Preferred Academic Credentials

• MS/PhD (Computer Science, Computer Engineering, Electrical Engineering, or related equivalent)

 Location:
San Jose, California, USA

#LI-G11 

#LI-HYBRID

At AMD, your base pay is one part of your total rewards package.  Your base pay will depend on where your skills, qualifications, experience, and location fit into the hiring range for the position. You may be eligible for incentives based upon your role such as either an annual bonus or sales incentive. Many AMD employees have the opportunity to own shares of AMD stock, as well as a discount when purchasing AMD stock if voluntarily participating in AMD’s Employee Stock Purchase Plan. You’ll also be eligible for competitive benefits described in more detail here.

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law.   We encourage applications from all qualified candidates and will accommodate applicants’ needs under the respective laws throughout all stages of the recruitment and selection process.