PyTorch Engineer

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. Graphcore’s teams include AI research specialists, silicon designers, software engineers, and systems architects with a culture of continuous learning and innovation.

Essential:

• Demonstrable strong software engineering skills.
• Experience in Python development.
• Experience developing performant C++ applications in a commercial setting.
• At least one of the following: experience writing ML kernels, experience in compiler development, experience with use and/or development of ML Frameworks, experience in a computationally intensive engineering field.

Desirable:

• Demonstrable knowledge of AI/ML.
• Understanding of computing architectures.
• Experience maintaining and supporting a complex code base.
• Experience in profiling and optimising high-performance code.

• Designing and implementing new features and maintaining and supporting existing features.
• Developing and maintaining unit tests, component tests, and integration tests.
• Optimising the software stack for performance.
• Managing complex technical tasks with cross-team dependencies.
• Managing code quality, code complexity, and technical debt.
• Providing technical leadership and contributing to the team’s technical direction and processes.
• Contributing to documentation, including user manuals and tutorials.
• Conducting code reviews.
• Resolving regressions, performance issues, and software defects.
• Coaching and mentoring other team members.
• Contributing to a collaborative team culture.
• Contributing to continuous improvements in ways of working.
• Working with upstream open-source development teams.