Principal AI Virtualization Engineer – GPU/XPU Backends

Broadcom; Austin (TX); On-site; USD 127,100 - 226,000; Full time

Benefits offered by this job

Medical, dental, and vision plans

401(K) participation with company matching

Employee Stock Purchase Program (ESPP)

Paid sick leave and vacation time

Job summary

Broadcom is hiring a Principal Software Engineer in Austin, Texas, to lead initiatives in AI Virtualization technology. The engineer will focus on developing the AI Virtualization Stack and optimizing performance for machine learning workloads.

This role requires extensive experience with ML frameworks, strong programming skills in C++ and Python, and a problem-solving mindset. As part of a dynamic team, you will enhance virtual machine capabilities while ensuring adherence to coding standards and quality metrics.

The position offers a competitive salary, a comprehensive benefits package, and opportunities for professional growth.

Qualifications

• 12+ years of related experience or Master’s degree and 10+ years of experience.
• 5+ years of experience in ML framework/runtime development.
• Strong understanding and direct experience with ML frameworks.
• Excellent communication and collaboration skills.
• Ability to thrive in a fast-paced work environment.

Responsibilities

• Research, design, and develop the AI Virtualization Stack for the ESXi server.
• Implement and optimize ML frameworks to ensure high-performance execution.
• Troubleshoot and address bugs related to AI/ML acceleration.
• Develop and maintain technical documentation for features.

Skills

ML framework/runtime development

GPU/XPU backend engineering

C++ programming

Python programming

Problem-solving

Communication skills

Git version control

Education

Bachelor's degree in Computer Science or related field

Master's degree in related field

Tools

ML frameworks (PyTorch, JAX)

OpenXLA

CUDA

Job description

Broadcom is hiring a Principal Software Engineer in Austin, Texas, to lead initiatives in AI Virtualization technology. The engineer will focus on developing the AI Virtualization Stack and optimizing performance for machine learning workloads.

This role requires extensive experience with ML frameworks, strong programming skills in C++ and Python, and a problem-solving mindset. As part of a dynamic team, you will enhance virtual machine capabilities while ensuring adherence to coding standards and quality metrics.

The position offers a competitive salary, a comprehensive benefits package, and opportunities for professional growth.