Principal AI Virtualization Engineer (GPU/XPU) - Equity

Broadcom Inc. seeks a Principal Software Engineer to advance the AI Virtualization Stack for ESXi, enabling hardware-agnostic acceleration of AI/ML workloads in virtual machines. You will develop PyTorch and JAX backends with OpenXLA, optimize ML acceleration, and collaborate with internal and external teams to support multi-vendor GPUs and XPUs. The role emphasizes performance tuning (KV-caching, FlashAttention) and requires strong C++/Python skills, plus experience with ML frameworks and