Senior Staff Engineer - PyTorch Accelerator (W2 only)

Job Title: Senior Staff Engineer PyTorch Accelerator

Duration: 6+ Months

Location: San Jose, CA (Onsite)

Basic Qualifications:

• 5+ years of experience in open-source ecosystem development or related roles.
• Strong hands-on experience with PyTorch internals: ATen, the dispatcher, operator registration, autograd, tensor/device abstractions, and Python/C++ extensions.
• Strong C++ and Python skills, with a track record of contributing clean, well-tested code to large open-source projects (design discussions, code reviews, pull requests, CI debugging).
• Experience integrating ML frameworks with hardware accelerators, runtimes, compilers, kernels, or device drivers, and a good understanding of GPU/AI accelerator architecture (memory hierarchy, asynchronous execution, streams/events, profiling).
• Comfortable with PyTorch build/test workflows (Linux, CMake, GitHub, CI) and debugging across Python, C++, runtime, and hardware layers.
• Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field, or equivalent experience.

Preferred Qualifications:

• Recognized PyTorch contributor or maintainer, with changes accepted into the main project.
• Experience with PyTorch accelerator integration: PrivateUse1 backend development, OpenReg-style reference implementations, or out-of-tree backend maintenance.
• Experience with torch.compile (TorchDynamo, TorchInductor), MLIR/LLVM, Triton, or XLA, or optimizing large-scale LLM training (e.g. VeRL) and inference (e.g., vLLM, SGLang).