Principal AI Virtualization Engineer for GPUs/ESXi

Broadcom; California (MO); 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

Paid holidays and vacation time

Job summary

Broadcom is seeking a Principal Software Engineer to lead the development of the AI Virtualization Stack for its ESXi server product. You will be responsible for optimizing high-performance AI/ML workloads on GPUs and XPUs. The ideal candidate will have extensive experience in ML frameworks and programming languages like C++ and Python.

This position offers a competitive salary and comprehensive benefits including health plans and 401(K) matching. Join us to help shape the future of virtualization technology at Broadcom.

Qualifications

• 12+ years of experience for Bachelor's degree holders or 10+ years for Master's degree holders.
• 5+ years of experience in ML framework/runtime development.
• Strong understanding of ML frameworks and compiler technologies.

Responsibilities

• Develop the AI Virtualization Stack for ESXi server.
• Optimize backends using OpenXLA for AI/ML workloads.
• Enhance performance-critical sections of ML code.

Skills

ML framework/runtime development

CUDA

C++

Python

Git

Education

Bachelor's degree in Computer Science or related field

Master's degree

Tools

OpenXLA

PyTorch

JAX

Job description

Broadcom is seeking a Principal Software Engineer to lead the development of the AI Virtualization Stack for its ESXi server product. You will be responsible for optimizing high-performance AI/ML workloads on GPUs and XPUs. The ideal candidate will have extensive experience in ML frameworks and programming languages like C++ and Python.

This position offers a competitive salary and comprehensive benefits including health plans and 401(K) matching. Join us to help shape the future of virtualization technology at Broadcom.