PyTorch Engineer

PyTorch Engineer Overview

As a Software Engineer in the PyTorch team at Graphcore, you will play a crucial role in making Graphcore hardware seamlessly integrate with the machine learning frameworks used by engineers daily. Your responsibilities will include designing, implementing, optimizing, and supporting features across PyTorch, Triton, and related technologies. Your contributions will enhance the performance and compatibility of Graphcore hardware for machine learning engineers and researchers, ensuring they can work effectively without altering their preferred development processes.

You will be part of a SCRUM team within the Frameworks department, where collaboration, code reviews, documentation writing, and user support are key activities. The team emphasizes maintainable code, thoughtful design, and a culture of curiosity about complex systems. You will be encouraged to take ownership of your work, make practical decisions, and communicate closely with software, AI/ML, research, and hardware teams.

Key Responsibilities:

• Develop software that integrates Graphcore accelerators with leading machine learning frameworks.
• Design, implement, optimize, and support features across PyTorch, Triton, and related technologies.
• Enhance compatibility and performance for machine learning engineers and researchers using Graphcore hardware.
• Contribute to the upstream development of PyTorch and support the hybrid Python and C++ extension.
• Assist in integrating compiler backend capabilities with torch.compile.
• Engage in problem-solving and build expertise within a large open-source ecosystem.

Requirements:

• Demonstrated strong core software engineering skills and a collaborative approach to teamwork.
• Experience in developing Python libraries.
• Proven ability to build efficient C++ applications in a commercial environment.
• Solid understanding of linear algebra, floating-point arithmetic, or numerical analysis.
• Proficient in using Git and conducting code reviews in a professional setting.
• Willingness to independently explore large and unfamiliar open-source codebases.

Benefits:

• Flexible working arrangements to help balance personal and professional life.
• Generous leave policy to allow for rest and personal time.
• Retirement planning support with up to 5% matched pension contributions.
• Phantom equity options to share in the company’s success.
• Access to well-designed office spaces with complimentary food and an on-site barista.
• Income protection and life assurance for financial security.
• Customizable benefits package including options for private medical insurance and dental coverage.
• Additional optional benefits such as a cycle to work scheme and health cash plan.

Graphcore is committed to fostering an inclusive environment where diverse backgrounds and experiences are valued. As an equal opportunity employer, we encourage applicants from all walks of life and are happy to provide support during the interview process for those who need adjustments.

If you are eager to influence how PyTorch operates on Graphcore hardware, we invite you to apply and join us on this exciting journey.

Degree Requirement: Not Specified

Visa Sponsorship May be