Sr. Staff Software Development Engineer - Python, ML, data science, hardware optimization

AMD - San Jose, CA

Full-time - Senior

San Jose, CA

Computer and Electronic Product Manufacturing

About the position

The Sr. Staff Software Development Engineer at AMD is responsible for enhancing the performance of key applications and benchmarks, focusing on machine learning and hardware optimization. This role involves collaboration with software partner architects and engineers, customer interaction, and the development of technical collateral to promote AMD's AI Engine tool flows. The ideal candidate will possess strong software engineering skills, leadership abilities, and a passion for technology, contributing effectively in a fast-paced environment.

Responsibilities

• Engage with software partner architects and engineers to work on performance optimizations and develop benchmarking plans.
• Develop technical collateral such as presentations and whitepapers to position the ML compiler in target market segments.
• Co-develop joint partner demos and proof-points with partners.
• Interact with customers to help them converge on their critical design needs.
• Drive successful adoption of AMD AI Engine tool flows by collaborating with key customers and internal teams.
• Work with Tech Marketing and Applications Engineering to understand customer needs and architect new features.
• Stay informed of software and hardware trends, especially regarding algorithms and architecture.
• Collaborate with the Architecture team to design next-generation ML solutions.

Requirements

• Bachelor's or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent.
• Experience in machine learning algorithms and frameworks.
• Expertise with Deep Neural Network architectures like CNN, RNN, LLM, etc.
• Proficiency in AI frameworks such as Pytorch, Tensorflow, Caffe, etc.
• Knowledge of acceleration platforms like GPU, TPU, APU, AIE, FPGAs.
• Strong programming skills in C++ and/or Python.
• Strong analytical skills and excellent written/verbal communication skills.
• Experience in optimizing machine learning applications for target architectures.

Nice-to-haves

• Experience with quantization techniques is a plus.
• Team player with the ability to work across diverse and geographically distributed teams.

Benefits

• Employee stock purchase plan
• Annual bonus or sales incentive eligibility
• Competitive benefits package