Summary

This mid-level software engineering role at Graphcore focuses on connecting Graphcore accelerators with PyTorch, Triton and related machine learning frameworks. You will work on framework features, C++ and Python extensions, compiler backend integration and upstream open-source contributions; flexible working is offered, but the exact remote arrangement is not specified.

Responsibilities

• Design and implement pytorch framework features
• Optimise triton and related technologies
• Contribute to upstream pytorch development
• Support hybrid python and c++ extensions
• Integrate compiler backends with torch.compile
• Review code and write technical documentation
• Support framework users across engineering teams