PyTorch Engineer

📋 Description

• Make PyTorch feel native, powerful and production ready on Graphcore AI compute.
• Build software layer connecting Graphcore accelerators with ML frameworks; scale workloads.
• Improve performance, design features, support users pushing hardware limits.
• Maintain core code, review changes, contribute to docs and upstream.
• Deep technical ownership and influence on direction.
• Collaborate across teams and open source communities for reliability.

🎯 Requirements

• Strong software engineering skills for quality and delivery.
• Experience developing in Python.
• Experience building performant C++ applications in a commercial setting.
• Experience in ML kernels, compiler development, ML frameworks or HPC engineering.
• Ability to coordinate complex tasks with cross-team dependencies.
• Clear communication, ownership and curiosity about high-performance software.

🎁 Benefits

• Flexible working: Balance work and life with flexibility
• Comprehensive healthcare and dental cover
• Phantom equity: Share in Graphcore’s success
• Fitness and wellbeing support: Medicover gym card
• Financial protection: Death and disability insurance
• Retirement savings: 4% employer contribution