Principal AI Software Engineer

Advanced Micro Devices, Inc.
San Jose, United States of America
8 days ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Languages
English
Compensation
$ 240K

Job location

San Jose, United States of America

Tech stack

Artificial Intelligence
Compilers
Nvidia CUDA
Microarchitecture
Device Drivers
OpenCL
Performance Tuning
TensorFlow
System Software
Large Language Models
Gpu Programming
Information Technology
Stable Diffusion

Job description

AMD AI Group isseekinga highly influential technical leaderin theROCmsoftwareforNew Product Initiatives. This role requires driving innovation anddeveloping next generation productsacross AMD's broad portfolio, servingInstinct, Radeon, Ryzen, Embedded, Gaming and Autonomous Drivingproduct lines., * Hardware-Software Co-design: Collaborate across hardware architecture, compiler, math libraries,kerneland framework teams to influence future silicon features based on evolving AI workload trends.

  • Strong Execution: Deliver innovations and roadmap for AI software stack across all AMD products, ensuring AMDremainsthe platform of choice for top-tier AI customers.

  • Workload Performance Engineering: Lead the profiling, analysis, and tuning of large-scale models (LLMs, Diffusion, Multimodal, andMoE) to ensure "out-of-the-box" performance excellence on AMD hardware.

  • Ecosystem Innovation: Drive the development of advanced tools and frameworks for performance estimation, modeling, and automated reporting.

  • Customer Engagement: Partner with top customers andhyperscalersto understand their unique workload requirements and deliver tailored architectural wins and software optimizations.

  • Community &Open Source: Mentor and inspireotherengineersand contribute toROCmOpensource., AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD's "Responsible AI Policy" is available here.

Requirements

You are a strong technical leader whois able towork across organizations, align architecture with product goals, and influenceinnovation andexecution. You bring deepexpertisein at least one major area of the stack, while alsodemonstratingthe curiosity, adaptability, and technical breadth to learn new domains, grow their impact, and influence the broader ecosystem.

As a successful candidate, you are recognized for technical excellence, and the ability to deliverinnovationsthrough collaboration across silicon, system software, AI/ML frameworks, libraries, and application enablement teams., * Knowledge in GPU architectures, basic knowledge of CPU architecture

  • Experience in AI/ML software stackspanning compilers, kernels, runtime, libraries, models, frameworks, and performance optimization layers

  • Understanding of GPU programming such asROCm, CUDA, OpenCL,etc

  • Experience in hardware/software co-design, building high-performance products across the full product lifecycle.

  • Experience with operating systems (OS) and device driver development is a plus, * Undergraddegree required.Bachelor of Science, Masters, or PhD degree with emphasis in Electrical Engineering, Computer architecture, or Computer Science with relevant experience preferred

Benefits & conditions

$240,000.00/Yr.-$360,000.00/Yr.

About the company

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.

Apply for this position