R&D Staff Software Engineer

About the role

Broadcom is a global leader in semiconductor and infrastructure software solutions. As part of our commitment to innovation and excellence, our VMware subsidiary is dedicated to shaping the future of virtualization technology. We are seeking talented individuals to join the GPU Virtualization Team, which is responsible for integrating GPUs into the ESXi Operating System and providing acceleration for AI/ML and graphics applications running inside virtual machines. The GPU Virtualization Team is part of the VMware Cloud Foundation (VCF) Division, which enables readily deployable, easily managed solutions with GPUs to unleash the power of heterogeneous computing for modern applications. Job Summary: We are seeking an experienced Staff Software Engineer focused on developing and integrating our AI Virtualization Stack to provide hardware-agnostic acceleration for AI/ML workloads on Virtual Machines. This role is critical in enabling multi-vendor GPU and XPU support using ML compilation technologies.

Responsibilities

• Research, design, and develop software components to enable hardware-agnostic AI/ML acceleration for our ESXi server product.
• Work directly with GPU partners to integrate, test, and certify their Linux-based drivers and kernel components for use on our platform.
• Work on packaging and release of Driver components in line with Broadcom's established process.
• Troubleshoot and address bugs related to AI/ML acceleration functionality.
• Deliver software that meets the coding guidelines and quality standards set by the VCF.
• Develop and maintain technical documentation for delivered features.
• Work closely with the larger team, including virtual driver and device team, as well as external GPU/XPU vendors, to provide end-to-end support for ML frameworks.
• Stay up-to-date with the latest GPU/XPU hardware architecture and AI/ML compiler technologies.

Requirements

• Bachelor's degree in Computer Science or a related field with 8+ years of related experience, or a Master's degree with 6+ years of related experience.
• Deep understanding of the Linux GPU stack, including device drivers, kernel modules, and user-space components.
• Experience with C++ and Python programming languages.
• Strong problem-solving skills and ability to troubleshoot complex issues.
• Excellent communication and collaboration skills, specifically with external technology partners.
• Experience with version control systems such as Git.
• Ability to thrive in a fast-paced and dynamic work environment.
• Familiarity with enterprise coding standards and best practices.
• Experience with ML frameworks (PyTorch, JAX) and graph/ML compiler technologies (e.g., OpenXLA).
• Experience with build infrastructure using Bazel, Make, Artifactory, etc.
• Experience integrating partner software into products.
• Must have legal authorization to work in the US
• Additional Job Description:
• Compensation and Benefits
• The annual base salary range for this position is $ 109,700.00 - 195,000.00
• This position is also eligible for a discretionary annual bonus in accordance with relevant plan documents, and equity in accordance with equity plan documents and equity award agreements.
• If you are located outside USA, please be sure to fill out a home address as this will be used for future correspondence.

Benefits

Additional Information

Please Note: 1. If you are a first time user, please create your candidate login account before you apply for a job. (Click Sign In > Create Account) 2. If you already have a Candidate Account, please Sign-In before you apply. Job Description:

Your Match

How well this role fits your profile.

Company Intel

What employees say

Worked at Broadcom? Share your experience