R&D Principal Software Engineer

Broadcom

Confirmed live 2 days ago High trust

Quick summary

Work type
On-site
Location
CAAustin, TX
Salary
$127,100–$226,000 / yr
Posted
87 days ago
Freshness
Confirmed live 2 days ago
Closes
Dec 11, 2026
Nearby
99+ roles within 25 mi

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $198k
This role $177k
$109k most similar roles pay here $293k

This role pays less than 63% of similar roles. Most pay $174,600–$220,800 — the shaded band above. At the midpoint, this role pays about $177k versus about $198k for comparable roles.

Based on 240 similar postings.

Employer

About Broadcom

Broadcom is a global semiconductor and infrastructure software company that designs and markets a wide range of networking, storage, and wireless connectivity solutions. Industry: Semiconductors & Infrastructure Software

Broadcom currently has 119 open roles on FindRole.

Listed pay typically runs $121,900–$195,000 across 105 roles with salary data.

Most-posted roles

View all roles at Broadcom

At a glance

TL;DR · R&D Principal Software Engineer

Role Posting Log in to save

As an R&D Principal Software Engineer within the GPU Virtualization Team, you will develop and integrate the AI Virtualization Stack for the ESXi server product. You will be responsible for researching, designing, and developing hardware-agnostic acceleration for AI/ML workloads on virtual machines, specifically focusing on multi-vendor GPU and XPU support using ML compilation technologies. Your daily work involves implementing and optimizing PyTorch and JAX backends via the OpenXLA framework, re-architecting performance-critical sections like KV-caching and FlashAttention, and troubleshooting complex bugs. You will utilize C++, Python, and Git while collaborating with internal teams and external vendors to support ML frameworks. The role addresses the technical challenge of providing high-performance execution for large language model inference across diverse hardware architectures within a virtualized environment.

Skills

PyTorch JAX OpenXLA C++ Python CUDA ROCm vLLM Triton ESXi GPU Virtualization ML Compilation FlashAttention KV-caching Git Kernel Development

What you'll do

  • Research, design, and develop the AI Virtualization Stack for the ESXi server product.
  • Implement and optimize PyTorch and JAX backends using the OpenXLA framework.
  • Re-architect performance-critical sections of ML acceleration code, including KV-caching and FlashAttention.
  • Troubleshoot and resolve bugs related to AI/ML acceleration functionality.
  • Develop and maintain technical documentation for all delivered features.
  • Provide end-to-end support for ML frameworks in coordination with internal teams and external vendors.
  • Monitor and integrate the latest GPU/XPU hardware architectures and ML compiler technologies.

What we're looking for

  • Bachelor's degree in Computer Science or related field and 12+ years of experience, or a Master's degree and 10+ years of experience.
  • At least 5 years of experience in ML framework/runtime development and GPU/XPU backend engineering.
  • Direct experience with ML frameworks such as PyTorch and JAX.
  • Experience with graph/ML compiler technologies like OpenXLA.
  • Proficiency in C++ and Python programming languages.
  • Experience with version control systems such as Git.
  • Legal authorization to work in the United States.

Related searches

More R&D Principal Software Engineer jobs All jobs at Broadcom

More like this

Similar roles

Principal Software Engineer

CVS Health

Remote (Massachusetts) +4 63 days ago $144,200–$288,400
Python GCP Vertex AI LLM MLOps Microservices RAG Vector Search React Angular CI/CD Pub/Sub BigQuery GKE Cloud Run FHIR HL7 SRE
10+ yrs exp Remote
Save

Principal Software Engineer

Autodesk

Atlanta, GA 29 days ago $139,000–$249,260
Distributed Systems Data Pipelines NoSQL Serverless API Design ETL Cloud Platform Schema Validation Search Relevance
8+ yrs exp Hybrid
Save

Principal Software Engineer

Oracle

73 days ago $114,600–$234,600
C Distributed Systems Cloud Architecture Python Kernel Programming Network Programming Unix Linux Elastic File System OCI HPC AI/ML
6+ yrs exp
Save

Principal Software Engineer

General Dynamics

San Antonio, TX 55 days ago $119,000–$161,000
DevSecOps CI/CD Kubernetes Docker Terraform Ansible Puppet Chef Python Java Go Bash AWS Azure Google Cloud Agile Methodology infrastructure-as-code OWASP Nessus Burp Suite Splunk
10+ yrs exp Hybrid
Save