Senior Software Engineer - PyTorch
GraphcoreGdańsk, Pomeranian Voivodeship, Poland · Posted 1 month agoDescription
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.
Similar jobs
Est. 80,000 GBP
About the job Help make Graphcore hardware feel native inside the ML frameworks engineers use every day. As a Software Engineer in our PyTorch team, you will help build the software that connects Graphcore accelerators w…
Est. 412,550 PLN
Salary Range: PLN 350,700 - 474,400 PLN Subject to alignment to the responsibilities and duties of the role About the job Build the AI compute libraries that turn next-generation hardware into real performance. As a Staf…
Est. 120,000 PLN
About the job Build the framework support that helps AI developers unlock Graphcore hardware. You will help Graphcore accelerators work seamlessly with state-of-the-art ML frameworks, including Triton and PyTorch. Report…
Est. 306,300 PLN
Salary Range: PLN 260,400 - 352,200 Subject to alignment to the responsibilities and duties of the role. About the job Help make Graphcore’s AI accelerators feel native inside modern Kubernetes environments. As a Senior…
Est. 306,300 PLN
Salary Range: PLN 260,400 - 352,200 Subject to alignment to the responsibilities and duties of the role About Graphcore At Graphcore, we’re building the future of AI compute. We’re a team of semiconductor, software and A…
Est. 306,300 PLN
Senior: PLN 260,400 - 352,200Staff: PLN 350,700 - 474,400Subject to alignment to the responsibilities and duties of the role - we currently have multiple positions available at both Senior and Staff levelAbout Graphcore…
Est. 412,550 PLN
Salary Range: PLN 350,700- 474,400 Subject to alignment to the responsibilities and duties of the role.Location: Gdańsk - Candidates must be local due to hybrid working policy About the job Make AI accelerator hardware f…
Est. 412,550 PLN
Salary Range: PLN 350,700 - 474,400 + Benefits + Equity Subject to alignment to the responsibilities and duties of the role. About the job Turn Graphcore’s infrastructure telemetry into intelligence that predicts issues…
Est. 80,000 GBP
About the job Build the framework support that helps AI developers unlock Graphcore hardware. You will help Graphcore accelerators work seamlessly with state-of-the-art ML frameworks, including Triton and PyTorch. Report…
Est. 95,000 GBP
About the job Build the compute kernels that make next-generation AI hardware perform at its limit. As a Senior Software Engineer, you will create high-performance AI compute libraries for Graphcore’s next-generation har…
Est. 412,550 PLN
Salary Range: PLN 350,700 - 474,400 Subject to alignment to the responsibilities and duties of the role. Location: Gdańsk - Hybrid Working Policy - 2-3 Days per week in office About the job Build the quality foundations…
Est. 306,000 PLN
Salary Range: PLN 260,000-352,000 Subject to alignment to the responsibilities and duties of the role About Graphcore At Graphcore, we’re building the future of AI compute. We’re a team of semiconductor, software and AI…
Est. 306,000 PLN
Salary Range: PLN 260,000 - 352,000 PLN Subject to alignment to the responsibilities and duties of the role About Graphcore At Graphcore, we’re building the future of AI compute. We’re a team of semiconductor, software a…
Est. 412,550 PLN
Salary Range: PLN 350,700 - 474,400 + Benefits + Equity Subject to alignment to the responsibilities and duties of the role. Location: Gdańsk - Hybrid Working Policy - 2-3 Days per week in office About the job Explore ho…
Est. 120,000 PLN
About the job Own the test strategy that proves Graphcore’s next generation drivers are production-ready. As a Staff System Software Engineer, you will shape how Graphcore validates complex device drivers and platform to…
Est. 230,500 PLN
Salary Range: PLN 196,000 - 265,000 Subject to alignment to the responsibilities and duties of the role About Graphcore At Graphcore, we’re building the future of AI compute. We’re a team of semiconductor, software and A…
Est. 412,000 PLN
Salary Range: PLN 350,000 - 474,000 PLN Subject to alignment to the responsibilities and duties of the role About Graphcore At Graphcore, we’re building the future of AI compute. We’re a team of semiconductor, software a…
Est. 80,000 GBP
About the job Validate the ML stack that turns accelerator hardware into trusted AI performance. This role sits where modern ML models meet Graphcore’s software and hardware stack. You will test, benchmark and validate c…
Est. 80,000 GBP
About the job Validate the ML stack that turns accelerator hardware into trusted AI performance. This role sits where modern ML models meet Graphcore’s software and hardware stack. You will test, benchmark and validate c…
Est. 80,000 GBP
About the job Validate the ML stack that turns accelerator hardware into trusted AI performance. This role sits where modern ML models meet Graphcore’s software and hardware stack. You will test, benchmark and validate c…
Est. 175,000 USD
About us 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…
Est. 120,000 PLN
About 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 breakthroug…
Est. 65,000 GBP
About the job Lead the projects that turn advanced hardware designs into products ready for manufacturing. As a Project Manager, you will drive hardware manufacturing projects from planning through production release. Yo…
Est. 86,350 GBP
Senior: GBP 73,500 - 99,500 Staff: GBP 97,300 - 131,700 Subject to alignment to the responsibilities and duties of the role - we currently have multiple positions available at both Senior and Staff level. About the job B…
Est. 80,000 GBP
About the job Build the cloud platform that puts Graphcore AI systems into real use. As a Senior Cloud Engineer, you will develop and operate private cloud services for Graphcore teams and users. You will turn product an…
Est. 86,350 GBP
Senior: GBP 73,500 - 99,500 Staff: GBP 97,300 - 131,700 Subject to alignment to the responsibilities and duties of the role - we currently have multiple positions available at both Senior and Staff level. About the job B…
Est. 86,350 GBP
Senior: GBP 73,500 - 99,500 Staff: GBP 97,300 - 131,700 Subject to alignment to the responsibilities and duties of the role - we currently have multiple positions available at both Senior and Staff level. About the job B…
Est. 120,000 GBP
About the job Build the cloud platform that powers Graphcore’s next generation AI systems. As a Principal Cloud Engineer, you will help develop, deploy and operate cloud services across Graphcore’s AI infrastructure. You…
Est. 412,550 PLN
Salary Range: PLN 350,700 - 474,400 + Benefits + Equity Subject to alignment to the responsibilities and duties of the role. Location: Gdańsk - Hybrid Working Policy - 2-3 Days per week in office About Graphcore At Graph…
Est. 80,000 GBP
About Graphcore At Graphcore, we’re building the future of AI compute.We’re a team of semiconductor, software and AI experts, with deep experience in creating the complete AI compute stack - from silicon and software to…