Software Engineer Kernel Programming Model

You will design PyTorch-native integration layers and runtime environments for custom kernels. You will implement virtual ISA abstractions, build programming tools and reference implementations, and support a kernel development ecosystem.

Responsibilities

• Design integration layers and runtime environments for custom kernels in the PyTorch ecosystem
• Implement virtual ISA abstractions for direct hardware control and tensor-level programmability
• Build programming tools, technical specifications, and reference implementations for kernel development

Requirements

• Experience with systems programming in Rust, C++, or Go
• Understanding of computer architecture, including ISA, SIMD, and memory hierarchies
• Experience designing and building programming interfaces
• Bachelor's degree in Computer Science or a related technical field, or equivalent practical experience
• Knowledge of low-latency asynchronous execution, kernel optimization, or hardware-software co-design
• Knowledge of programming language theory and compiler infrastructure, including LLVM or MLIR
• Experience with developer communities, open source, technical evangelism, or developer ecosystems