PyTorch Engineer

About the Company

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.

Responsibilities

• Designing and implementing new features and maintaining existing features
• Developing and maintaining unit, component, and integration tests
• Optimising the software stack for performance
• Managing complex technical tasks and cross-team dependencies
• Managing code quality, complexity, and technical debt
• Providing technical leadership and contributing to team direction
• Contributing to documentation, user manuals, and tutorials
• Conducting code reviews and resolving regressions or defects
• Coaching and mentoring team members
• Working with upstream open-source development teams

Requirements

• Strong software engineering skills
• Experience with Python development
• Experience developing performant C++ applications in a commercial setting
• Experience in at least one of: writing ML kernels, compiler development, ML framework development, or computationally intensive engineering

Preferred Qualifications

• 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