Broadcom

R&D Principal Software Engineer

Broadcom$127K — $226K *
Information Technology
8 - 10 years of experience
Job Overview by Ladders

Qualifications

  • Bachelor's in Computer Science or related field with 12+ years of experience, or Master's with 10+ years
  • 5+ years of experience in ML framework/runtime development and GPU/XPU backend engineering
  • Strong understanding of ML frameworks like PyTorch and JAX, and ML compiler technologies such as OpenXLA
  • Proficiency in C++ and Python programming
  • Excellent problem-solving skills and troubleshooting abilities
  • Strong communication and collaboration skills
  • Familiarity with version control systems like Git
  • Ability to function in a fast-paced environment

Responsibilities

  • Research and develop the AI Virtualization Stack for ESXi server
  • Implement and optimize PyTorch and JAX backends using OpenXLA
  • Analyze and improve performance-critical ML acceleration code
  • Troubleshoot issues related to AI/ML acceleration
  • Ensure software meets coding guidelines and quality standards
  • Create and maintain documentation for features
  • Collaborate with virtual driver and device teams and external GPU/XPU vendors
  • Stay updated on GPU/XPU hardware and AI/ML compiler technologies

Benefits

  • Discretionary annual bonus
  • Competitive new hire equity grant and annual equity awards
  • Medical, dental, and vision plans
  • 401(K) participation with company matching
  • Employee Stock Purchase Program (ESPP)
  • Employee Assistance Program (EAP)
  • Paid holidays, sick leave, and vacation time
Full Job Description

Please Note:

1. If you are a first time user, please create your candidatelogin 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:

Job Summary:

We are seeking an experienced Principal Software Engineer who has experience leading initiatives in the past. As a Principal Engineer, you will be 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 the AI Virtualization Stack for our ESXi server product.

  • Implement and optimize PyTorch and JAX backends using the OpenXLA framework to ensure high-performance AI/ML workload execution across GPUs and XPUs.

  • Analyze and re-architect performance-critical sections of the ML acceleration code, focusing on optimization techniques for LLM inference such as KV-caching and FlashAttention.

  • 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 related field and 12+ years of related experience or Masters degree and 10+ years of related experience.

  • 5+ years of experience in ML framework/runtime development, GPU/XPU backend engineering.

  • Strong understanding and direct experience with ML frameworks (PyTorch, JAX) and graph/ML compiler technologies (e.g. OpenXLA).

  • Experience with C++ and Python programming languages.

  • Strong problem-solving skills and ability to troubleshoot complex issues.

  • Excellent communication and collaboration skills.

  • 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 inference servers such as vLLM, Triton.

  • Experience with low-level GPU kernel development and writing custom kernels (e.g., CUDA, ROCm, or similar).

Must have legal authorization to work in the US

Additional Job Description:

Compensation and Benefits

The annual base salary range for this position is$127,100 - $226,000.

As a valued member of our team, you'll be eligible for a discretionary annual bonus and the opportunity to receive not only a competitive new hire equity grant, but also annual equity awards, connecting your success directly to the company's growth. All subject to relevant plan documents and 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.

If you are located outside USA, please be sure to fill out a home address as this will be used for future correspondence.

About Broadcom

Broadcom Limited is a designer, developer and global supplier of a range of analog semiconductor devices with a focus on III-V based products. Its product portfolio is extensive and includes over 6,500 products that it sells into four primary target markets: wireless communications, wired infrastructure, industrial and automotive electronics, and consumer and computing peripherals. Applications for its products in these target markets include cellular phones, consumer appliances, data networking and telecommunications equipment, enterprise storage and servers, renewable energy and smart power grid applications, factory automation, displays, optical mice and printers. Its product portfolio ranges from simple discrete devices to complex sub-systems that include multiple device types and incorporate firmware for interface between digital systems. It also includes mechanical hardware that interfaces with optoelectronic or capacitive sensors.

Broadcom Careers

Join Broadcom—a leader in technological innovation and a beacon for professionals seeking transformative career opportunities. At Broadcom, we are committed to building a diverse team of talented individuals who are ready to drive innovation and lead the industry forward. Work You’ll Do At Broadcom, you will be part of a dynamic team that thrives on innovation and leadership. Our professionals are at the forefront of developing cutting-edge technologies that redefine how we connect and communicate. Your work will not only influence the direction of technology but also contribute to industries and communities globally. Broadcom offers a range of job opportunities that cater to different skills and career aspirations. From engineering to marketing, our positions are designed to challenge your professional growth while contributing significantly to our global success. Join Our Inclusive Culture Broadcom is dedicated to fostering an inclusive culture where diversity is celebrated, and every team member is valued. We believe that diverse perspectives lead to more innovative solutions. Our commitment to diversity training ensures that all employees have the opportunity to thrive. Internship Programs Start your career with a Broadcom internship. Our internships provide invaluable industry experience and a chance to develop essential skills in a real-world setting. Interns at Broadcom are considered integral members of the team and are given responsibilities that challenge and enhance their academic learning. Professional Growth and Benefits Broadcom is deeply invested in the professional growth of its employees. We offer comprehensive benefits and resources to support both your professional and personal development. This includes competitive employment benefits, leadership training, and opportunities for career advancement. Networking and Career Development We encourage our employees to engage in networking and professional development activities that enhance their career paths. Broadcom’s expansive network of industry professionals provides a robust environment for growth and innovation. Employees are supported in their efforts to pursue certifications, attend workshops, and take part in leadership training to hone their expertise and advance their careers. Hiring Process Our hiring process is designed to identify and attract professionals who are passionate, curious, and driven. Starting with your resume submission, through to the interview stage, each step is an opportunity for us to learn about your strengths and for you to understand our company's values and vision. Explore Job Opportunities Whether you are seeking an entry-level position or a more senior role, Broadcom offers a range of opportunities to build a rewarding career. Explore the positions available and find where your skills and interests align with our needs. Stay Connected Join the Broadcom team and be part of a company that values innovation, leadership, and a commitment to quality. Search open positions that match your skills and interests. We look for individuals who are creative, solution-driven, and ready to contribute to our team’s success. SEARCH BROADCOM JOBS Keep Up to Date Stay ahead with career tips, insider perspectives, and industry-leading insights you can put to use today—all from the people who work here. READ CAREERS BLOG Job Alert Emails Personalize your subscription to receive job alerts, latest news, and insider tips tailored to your preferences. Discover the exciting and rewarding opportunities that await at Broadcom. Join Broadcom and propel your career to new heights in an environment where your contributions lead to real-world solutions and success.
Learn more about Broadcom
Size
20,000 employees
Market Cap
$231.5 billion
Industry
Net Income
$3.9 billion
Founded
1991
5 Year Trend
+13.5%
Revenue
$24.6 billion
NASDAQ

Similar Jobs

More Jobs at Broadcom

More Information Technology Jobs

Find similar R&D Principal Software Engineer jobs: