PyTorch Engineer
GraphcoreAI & company
Gdańsk, PolandUSD 260,400 - 352,200Mid
Baillie Gifford
SoftBank Group
Ontario Teachers' Pension Plan
Fidelity International
Schroders
Draper Esprit
Data & AI
About the role
TL;DR
Develop and optimize software for Graphcore's machine learning accelerators in PyTorch.
- •Join our PyTorch team to ensure Graphcore's machine learning accelerators have first-class support in PyTorch and Triton.
- •Key Responsibilities Design, implement, and optimize software for machine learning accelerators.
- •Collaborate with other engineers to deliver new features and optimize performance.
- •Review code changes, write technical documentation, and support users.
- •Contribute to the team's technical direction and processes.
- •Work with upstream communities and maintain the code base.
- •Requirements Strong background in machine learning frameworks, particularly PyTorch.
- •Experience with optimizing performance and managing code quality.
- •Excellent communication and collaboration skills.
- •Pro-active attitude and ability to work on open-ended technical tasks.
- •Flexibility to work on other frameworks as needed.
Required skills
PyTorchTensorFlowCI/CDDockerKubernetes
Nice-to-have skills
JAXEngineering ManagementTechnical LeadershipAgileScrum
Domain expertise
ai
Tech stack
PyTorchJAXTensorFlowCI/CDDockerKubernetes