Sr. Product Development Engineer - AI/ML Frameworks, Python

AMD; San Jose (CA); On-site; USD 125,000 - 189,000; Full time

Job summary

Join a leading tech company as a Senior Product Development Engineer focusing on advancing AI/ML frameworks and solutions. In this role, you'll engage heavily with customers and cross-functional teams, evaluate and improve ML inferencing pipelines, and produce valuable educational content. The ideal candidate will possess strong communication skills and a background in embedded device optimization with a focus on AI technologies.

Benefits

Competitive benefit packages

Inclusive employer culture

Qualifications

• Experience with AI frameworks such as PyTorch and ONNX.
• Proven experience deploying and optimizing ML models on embedded hardware.
• Ability to present features, solutions clearly and concisely.

Responsibilities

• Facilitate customer onboarding and delivery of best-in-class customer experience.
• Conduct performance benchmarking and bottleneck analysis.
• Develop use cases, examples, and tutorials for AI inference software.

Skills

AI frameworks

Communication

Analytical skills

Problem-solving

Education

Bachelor’s or Master’s in Computer Engineering, Computer Science, or Electrical Engineering

Tools

Python

C

C++

Git

Docker

Job description

Join to apply for the Sr. Product Development Engineer - AI/ML Frameworks, Python role at AMD

Join to apply for the Sr. Product Development Engineer - AI/ML Frameworks, Python role at AMD

Get AI-powered advice on this job and more exclusive features.

This range is provided by AMD. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more.

Base pay range

$125,440.00/yr - $188,160.00/yr

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 looking for a highly motivated product engineer to own and manage the customer experience for AMD’s Software Solutions for ML inferencing. The successful candidate will:

• Facilitate adoption of AMD’s ML inferencing solutions by facilitating customer on-boarding and delivering best-in-class customer experience
• Work with R&D, marketing, and key customers to drive the evolution of the product
• Support strategic business engagements leveraging AMD’s ML inferencing solutions
  
  
  

We have competitive benefit packages and an award-winning culture. Join us!

The Person

The ideal candidate will be capable of communicating problems, solutions, and project status up to the executive level. You have the capability assuming the consultative or senior level responsibilities for large projects or product development initiatives, while dealing with internal and external groups on behalf of the group or project. This role allows the opportunity to influence technical decisions that have a significant impact on multiple products or the product line.

Key Responsibilities

• Evaluate end-to-end ML inferencing pipelines optimized for AMD’s Embedded devices. This role involves conducting performance benchmarking, memory profiling, and bottleneck analysis in addition to extensive flow, and usability, analysis for model inferencing flows.
• Develop use cases, examples, tutorials, and methodology collaterals for AI inference software components and ecosystems, including Compiler, Quantizer, Optimizer, Runtime, Profiler, Visualizer, Debugger, and more targeting embedded devices.
• Engage with engineering teams to collaborate on product development, analyze product/feature specifications and usability to provide early feedback, and understand product/feature usage holistically to identify potential customer pain points.
• Interact with internal and external customers to understand their issues, assist them in debugging their workflows to meet critical model and flow-specific requirements, and create use cases to reproduce issues, driving successful adoption.
• Collaborate with the sales and marketing teams on strategic business engagements.
• Work closely with R&D to prioritize issues and manage escalations.
  
  
  

Preferred Experience

• Experience working with AI frameworks such as PyTorch and ONNX.
• Proven experience deploying and optimizing ML models on embedded hardware(microcontrollers, custom SOCs or similar platforms)
• Work experience with state-of-the-art models in one or more domains: CNN, Transformers, LLMs, Generative AIs.
• Experience in model analysis (CNN, LLM), performance, benchmarking, and accuracy testing.
• Experience in quantizing and pruning ML networks is preferred.
• Experience working on and managing open-source AI projects is preferred.
• Hands-on experience with Python, C, and C++.
• Expertise with popular development tools such as Git, CMake, Make, Conda, Docker, VSCode, bash, Linux.
• Demonstrated expertise in performance analysis and debugging across software/hardware boundaries
• Ability to think from the end-user's perspective, understand their needs, and advocate for them.
• Documentation skills with the ability to present features, solutions clearly and concisely.
• Excellent analytical, problem-solving, and communication skills
  
  
  

Academic Credentials

• Bachelor’s or Master’s in computer engineering or computer science or electrical engineering, or comparable disciplines
  
  
  

LOCATION:

• San Jose, CA OR
• Longmont, CO
  
  
  

Benefits offered are described: AMD benefits at a glance.

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.

Seniority level

• Seniority level

  Mid-Senior level

Employment type

• Employment type

  Full-time

Job function

• Industries

  Semiconductor Manufacturing

Referrals increase your chances of interviewing at AMD by 2x

Get notified about new Senior Product Development Engineer jobs in San Jose, CA.

Sr. Product Engineer, Tesla for Business

Senior Research And Development Engineer

Senior Mechanical Design Engineer - Fuel Systems

Menlo Park, CA $126,000.00-$148,000.00 3 weeks ago

Sr. R&D Engineer (Process Development) - Shockwave

Sr. Product Engineer, Identity Management

Senior Mechanical Engineer - Low Voltage Electronics

Sr Product Engineer, Growth Technology - Performance Marketing

San Jose, CA $108,800.00-$149,600.00 2 weeks ago

Senior Mechanical Engineer – Interconnect Design

Sr Product Cost Engineer, Amazon Custom Modules

Senior Software Engineer - Cortex Apps (LLM Products)

Menlo Park, CA $195,000.00-$287,500.00 2 weeks ago

Senior Compliance Engineer, EMC, AWS Product Compliance Engineering

Sr. Optical Network Engineer - Silicon Photonics, Network Product Development - Optics

Senior Mechanical Product Design Engineer, Google Cloud

Sunnyvale, CA $147,000.00-$216,000.00 3 days ago

Senior Mechanical Engineer, Advanced Engineering

Senior Mechanical Product Design Engineer

Santa Clara, CA $133,000.00-$185,000.00 2 days ago

Senior Mechanical Engineer, Advanced Energy

Senior Product Development Engineer for IST

Senior Mechanical Design Engineer, Advanced Process Technologies (Battery)

Palo Alto, CA $180,000.00-$215,000.00 2 weeks ago

Santa Clara, CA $132,000.00-$253,000.00 3 days ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.