PyTorch Integration Engineer for Accelerated ML

Graphcore is seeking a Software Engineer to help make Graphcore hardware feel native inside ML frameworks. You will design, implement, and optimize features that connect Graphcore accelerators with PyTorch, Triton and related technologies.

Join a SCRUM team in Frameworks, contribute to upstream PyTorch, and support a hybrid Python/C++ extension. You will solve real engineering problems, ship tooling, and improve performance and developer experience for ML researchers.