Staff Compiler Engineer - PyTorch + Kernel DSLPLATE

A Senior Staff or Principal Compiler Engineer role within the AGI Computing Lab focuses on adapting PyTorch's torch.compile and building custom kernel domain-specific languages (DSLs) for scalable AI/ML workloads. The position involves designing compiler passes, placement strategies, and parallelism-aware lowering to optimize performance across diverse hardware backends while minimizing energy consumption. Collaboration with software and hardware engineers is essential to address complex system-level challenges in non-uniform memory hierarchies.