R&D Staff Software Engineer
About The Position
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. 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.
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.
- Must have legal authorization to work in the US
Nice To Haves
- 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.
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.
Benefits
- Medical, dental and vision plans
- 401(K) participation including company matching
- Employee Stock Purchase Program (ESPP)
- Employee Assistance Program (EAP)
- company paid holidays
- paid sick leave and vacation time
- Paid Family Leave and other leaves of absence
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
