Senior Software Engineer - PyTorch

Salary Range: PLN Range: 260,400 - 352,200 

Subject to alignment to the responsibilities and duties of the role

About the job

Make PyTorch work brilliantly on hardware built for the next era of machine intelligence.

As a Senior Software Engineer in our PyTorch team, you will help Graphcore hardware feel natural to ML engineers and researchers. You will design, optimise and support the software that connects our accelerators with widely used frameworks.

Your work will shape upstream PyTorch contributions, custom kernels and compiler backend integration with torch.compile. The goal is simple: better compatibility, stronger performance and an easier developer experience.

You will work across Python, C++ and complex open source codebases. You will also help coordinate technical tasks, review code and improve how the team builds.

This is a role for engineers who want deep ownership and real technical stretch. You will grow expertise in PyTorch while influencing systems used by ambitious ML teams.

The team and culture

You will join the PyTorch team within Frameworks, working in a SCRUM team that ships features, fixes issues and improves performance. Work happens through open discussion, code review and close contact with software, AI/ML, research and hardware teams.

Engineers are trusted to take responsibility for complex, open-ended tasks. Decisions are shaped by technical evidence, user impact and healthy challenge.

The team values clarity, learning and practical progress. You will mentor others, speak up when quality matters and help improve how the team works.

What we’re looking for

· Strong software engineering skills across complex systems.
· Experience building in Python and performant C++ in a commercial setting.
· Confidence with linear algebra and floating-point arithmetic.
· Strong git skills, including code review in a commercial engineering team.
· Ability to investigate unfamiliar, large or open source codebases and write clear technical documentation.
· Experience in at least one of ML kernels, compiler development, ML frameworks or computationally intensive engineering.

Benefits

· Flexible working: Balance your work and personal life with greater flexibility
· Comprehensive healthcare and dental cover: Premium medical and dental insurance, with the option to extend cover to your family
· Phantom equity: Share in Graphcore’s success
· Fitness and wellbeing support: A fully funded Medicover gym card, giving you access to a wide range of gyms, fitness classes and leisure centres
· Financial protection: Death and disability insurance, providing financial security for you and your loved ones
· Retirement savings: Build your financial future, with a 4% employer contribution from Graphcore, alongside your own contributions
· Employee Assistance Programme (EAP): Confidential advice and support for you and your family across a range of personal, financial and wellbeing matters

We welcome people from all backgrounds and experiences and are committed to building an inclusive environment where everyone can do their best work.
We’re an equal opportunity employer and recognise that everyone brings different strengths and perspectives. If you need any adjustments during the interview process, just let us know - we’re happy to support you.

Join the Team at Graphcore

Graphcore is one of the world’s leading innovators in Artificial Intelligence compute. It is developing hardware, software and systems infrastructure that will unlock the next generation of AI breakthroughs and power the widespread adoption of AI solutions across every industry.

As part of the SoftBank Group, Graphcore is a member of an elite family of companies responsible for some of the world’s most transformative technologies. Together, they share a bold vision: to enable Artificial Super Intelligence and ensure its benefits are accessible to everyone.

Graphcore’s teams are drawn from diverse backgrounds and bring a broad range of skills and perspectives. A melting pot of AI research specialists, silicon designers, software engineers and systems architects, Graphcore brings together deep expertise to solve complex problems and deliver meaningful progress in AI compute.

If you want to shape how PyTorch runs on Graphcore technology, we’d love to hear from you. Apply now to help build the software foundations for the next generation of AI.