Senior Software Engineer - PyTorch
Graphcore is a SoftBank-owned AI-compute company developing AI processors, systems, and software for machine-learning workloads.
Funding history
About Graphcore
Graphcore develops Intelligence Processing Units (IPUs) and the Poplar SDK for building and running machine-learning applications. It continues operating under the Graphcore name as a wholly owned SoftBank Group subsidiary.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will design, optimise, and support software connecting AI accelerators to widely used machine-learning frameworks. You will contribute upstream PyTorch changes, custom kernels, and torch.compile compiler-backend integration. You will review code, coordinate technical work, document unfamiliar codebases, and mentor others.
Requirements
- Software engineering experience with complex systems
- Python
- C++
- Linear algebra
- Floating-point arithmetic
- Git
- Code review
- Technical documentation
- Experience with ML kernels, compiler development, ML frameworks, or computationally intensive engineering
Responsibilities
- Design, optimise, and support software connecting accelerators to ML frameworks
- Contribute to upstream PyTorch
- Develop custom kernels
- Integrate compiler backends with torch.compile
- Coordinate technical tasks
- Review code
- Improve team development practices
- Mentor others
Benefits
- Flexible working
- Medical and dental insurance, with an option to extend cover to family
- Phantom equity
- Fully funded Medicover gym card
- Death and disability insurance
- 4% employer retirement contribution
- Employee Assistance Programme