Software Engineer – AI/ML Platform Solutions

Advanced Micro Devices, IncSan Jose, CA
4hOnsite

About The Position

At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you’ll discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges—striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond. Together, we advance your career. As a key member of the engineer team, you will bridge the gap between high-level AI frameworks and cutting-edge hardware. This role isn't just about writing code; it’s about architecting the software ecosystem that makes next-generation AI acceleration possible. You are a self-motivated engineer and problem-solver who thrives on the challenge of building at the bleeding edge of AI hardware. With deep technical mastery in compiler and embedded C, you possess the rare ability to navigate complex, heterogeneous systems and bridge the gap between high-level frameworks and bare-metal runtime. You are a proactive collaborator who doesn't wait for a perfect roadmap; instead, you excel at filling the gaps of incomplete documentation and driving architectural clarity to turn ambitious AI solutions into reality.

Requirements

  • Expert in C and Embedded systems development of ARM SoCs.
  • Experience in architecting, designing and implementing complex heterogenous or multi-processor system.
  • BS or MS in Computer Engineering, or related equivalent.

Nice To Haves

  • Hands-on MLIR/LLVM experience
  • Knowledge of AI/ML frameworks like Pytorch, Tensor flow, JAX, etc.
  • Knowledge of Acceleration template library like Cutlass, DNNL etc.
  • Knowledge of Acceleration platforms like GPU, TPU, APU, FPGAs

Responsibilities

  • Contributing to the architecture and design of the AI solutions platform.
  • Developing Bare Metal and/or Linux runtime for AI/ML acceleration engines.
  • Work with different teams to identify problems and create solutions
  • Work with management team on project planning activities.

Benefits

  • AMD benefits at a glance.
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service