Staff Software Engineer

Graphcore; Bristol; On-site; GBP 50,000 - 80,000; Full time

Job summary

A leading company in the machine learning industry seeks a framework engineer to enhance support for Graphcore hardware in popular ML frameworks like PyTorch. The role involves collaborating in a SCRUM team to deliver new features, optimize performance, and mentor junior engineers, all within an agile environment.

Benefits

Flexible working

Generous leave

Private medical insurance

Health cash plan

Dental plan

Pension (up to 5% match)

Life assurance

Income protection

Parental leave

Support programs

Qualifications

• Experience developing performant C++ applications in a commercial setting.
• Proficiency in Python development.
• Experience with ML frameworks like PyTorch.

Responsibilities

• Design and implement new features, maintain, and support existing features.
• Develop and maintain unit, component, and integration tests.
• Optimize the software stack for performance.

Skills

C++

Python

AI/ML concepts

Profiling and optimizing high-performance code

Job description

Reporting to a Team Lead in the frameworks team, you will play a pivotal role in designing, implementing, optimizing, maintaining, and supporting software to ensure Graphcore hardware receives first-class support in popular machine learning frameworks such as PyTorch.

As a framework engineer in the PyTorch team

You will be part of a SCRUM team working on delivering new features, including performance optimizations, code reviews, writing technical documentation, collaborating with upstream communities, maintaining the codebase, and supporting users.

You are expected to collaborate closely with other engineers, act as a role model for junior team members, and work with teams developing low-level libraries and ML applications on top of the frameworks.

The Team

The frameworks team ensures Graphcore hardware is seamlessly supported in user-facing machine learning frameworks such as PyTorch, Triton, Jax, and TensorFlow, focusing on ease-of-use and performance.

Our software team is at the forefront of the machine learning industry, fostering a culture of continuous improvement and personal development, all within an agile working environment.

Responsibilities and Duties

1. Design and implement new features, maintain, and support existing features.
2. Develop and maintain unit, component, and integration tests.
3. Optimize the software stack for performance.
4. Manage code quality, complexity, and technical debt.
5. Contribute to documentation, including user manuals and tutorials.
6. Conduct code reviews.
7. Resolve regressions, performance issues, and defects.
8. Coach and mentor team members.
9. Foster a collaborative team culture.
10. Contribute to continuous improvement processes.
11. Work with upstream open source development teams.

Candidate Profile

• Experience developing performant C++ applications in a commercial setting.
• Proficiency in Python development.
• Experience with or development of ML frameworks like PyTorch.
• Strong knowledge of AI/ML concepts.
• Experience in a computationally intensive engineering field.
• Understanding of compiler development.
• Knowledge of computing architectures.
• Experience maintaining complex codebases.
• Experience writing ML kernels.
• Skills in profiling and optimizing high-performance code.

In addition to a competitive salary, Graphcore offers flexible working, generous leave, private medical insurance, health cash plan, dental plan, pension (up to 5% match), life assurance, income protection, parental leave, and support programs. We promote an inclusive environment and provide reasonable adjustments for applicants.

Applicants must have the right to work in the UK; we cannot provide visa sponsorship at this time.