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 various teams and partners to develop innovative solutions and technical collateral, ensuring the successful adoption of AMD's AI Engine tool flows. The position requires a strong background in software engineering, machine learning algorithms, and a passion for technology.

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 accordingly.
• Stay informed of software and hardware trends, especially regarding algorithms and architecture.
• Collaborate with the Architecture team to design next-generation ML solutions.

Requirements

• Experience in machine learning algorithms and frameworks.
• Expertise with Deep Neural Network architectures like CNN, RNN, LLM, etc.
• Proficiency with 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.
• Excellent written and verbal communication skills.
• Experience in optimizing machine learning applications for target architectures.
• Familiarity with quantization techniques.

Nice-to-haves

• Experience working in cross-functional and geographically distributed teams.

Benefits

• Base pay based on skills and experience.
• Eligibility for annual bonuses or sales incentives.
• Opportunity to own shares of AMD stock through the Employee Stock Purchase Plan with discounts.
• Competitive benefits package.