R&D Staff Software Engineer
Top focus
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
About Us: 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
Qualifications
- 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
Nice To Have
- 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
Requirements
- 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
- equity in accordance with equity plan documents and equity award agreements.
- Broadcom offers a competitive and comprehensive benefits package: 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.
- The company follows all applicable laws for Paid Family Leave and other leaves of absence.
- Broadcom is proud to be an equal opportunity employer.
- We will consider qualified applicants without regard to race, color, creed, religion, sex, sexual orientation, national origin, citizenship, disability status, medical condition, pregnancy, protected veteran status or any other characteristic protected by federal, state, or local law.
- We will also consider qualified applicants with arrest and conviction records consistent with local law.
- If you are located outside USA, please be sure to fill out a home address as this will be used for future correspondence.