2026 Graduate Software Engineer - PyTorch
As a Graduate Software Engineer in the PyTorch team at Graphcore, you will be instrumental in designing, implementing, optimizing, and maintaining software that ensures Graphcore's machine learning accelerators are seamlessly integrated with leading frameworks like PyTorch and Triton. This role places you within a dynamic SCRUM team dedicated to delivering new features, enhancing performance, reviewing code changes, authoring technical documentation, collaborating with upstream communities, and providing user support.
Your primary responsibilities will include developing and maintaining unit, component, and integration tests, optimizing the software stack for enhanced user performance, managing code quality and complexity, and addressing technical debt. Additionally, you will contribute to user manuals and tutorials, conduct code reviews, resolve regressions and performance issues, and engage with upstream open-source development teams to foster continuous improvement.
Candidates should possess a Bachelor's or Master's degree in Computer Science, Mathematics, Machine Learning, Data Science, or a related field, along with experience in Python development. Preferred qualifications include experience in compiler development, utilization or development of machine learning frameworks, or work in computationally intensive engineering fields. Demonstrable knowledge of AI/ML, understanding of computing architectures, experience with complex codebases, proficiency in profiling and optimizing high-performance code, and experience in writing ML kernels are advantageous.
Graphcore offers a competitive salary, flexible working arrangements, generous annual leave, private medical insurance, a health cash plan, dental coverage, a pension scheme matched up to 5%, life assurance, and income protection. Additional benefits include a generous parental leave policy, an employee assistance program covering health and mental well-being, and access to healthy food and snacks at the central Bristol office, which features an in-house barista bar.
Joining Graphcore means becoming part of a diverse team committed to continuous learning and innovation in the AI industry. The company fosters a culture of collaboration, openness, and learning, providing opportunities for professional growth and the chance to make a significant impact in the field of machine learning.