Software Engineer - Triton

Graphcore is a leading innovator in Artificial Intelligence compute developing hardware, software, and systems infrastructure to enable the next generation of AI breakthroughs. The Frameworks team connects Graphcore hardware with ML frameworks like PyTorch, Triton, JAX, and TensorFlow, focusing on usability and performance.

• Strong software engineering skills with ability to manage quality, complexity, and technical debt.
• Experience developing in Python and C++.
• Experience in compiler development.
• Ability to manage complex technical tasks involving cross-team collaboration.
• Clear communication and proactive collaboration skills.
• Interest in AI, ML frameworks, performance optimization, or computationally intensive engineering.

Nice to have:

• Transferable skills and diverse experiences relevant to the role.

• Build and optimize framework support to integrate Graphcore hardware with ML frameworks including Triton and PyTorch.
• Design, implement, and maintain critical software components.
• Deliver new features and review code in the Triton team.
• Write technical documentation.
• Collaborate across Graphcore and with open-source communities.
• Coordinate technical work with cross-team dependencies.