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

AMD - San Jose, CA

Full-time - Mid Level

San Jose, CA

Computer and Electronic Product Manufacturing

About the position

AMD is seeking a specialized software engineer focused on enhancing the performance of key applications and benchmarks. This role involves collaboration with a talented team to leverage the latest hardware and software technologies, aiming to drive innovation in computing experiences across various sectors including data centers, AI, and gaming.

Responsibilities

• Engage with software partner architects and engineers for performance optimizations and 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 in 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.
• Ability to work across diverse teams and in geographically distributed teams.

Benefits

• Base pay dependent 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.