Staff Software Engineer Bristol, UK

graphcore; Bristol; On-site; GBP 40,000 - 80,000; Full time

Job summary

An established industry player is seeking a talented framework engineer to join their dynamic team in Bristol. This role involves designing and implementing features to ensure top-notch support for machine learning frameworks like PyTorch. You will collaborate closely with other engineers, mentor junior team members, and contribute to a culture of continuous improvement. The company offers a vibrant work environment focused on personal development and agile practices, making it an exciting opportunity for those passionate about cutting-edge technology in the machine learning sector.

Benefits

Flexible working

Private medical insurance

Health cash plan

Dental plan

Pension (matched up to 5%)

Life assurance

Income protection

Generous parental leave policy

Employee assistance programme

Healthy food and snacks

Qualifications

• Experience in developing performant C++ applications in a commercial setting.
• Proficient in Python development with a focus on machine learning.

Responsibilities

• Design and implement new features while maintaining existing ones.
• Optimize software stack for enhanced performance and manage code quality.
• Collaborate with upstream open source development teams.

Skills

C++ Development

Python Development

Machine Learning Frameworks

Performance Optimization

Code Review

Technical Documentation

Job description

Job Summary

Reporting to a Team Lead in the frameworks team, you will play a pivotal role in designing, implementing, optimising, maintaining, and supporting the software that ensures Graphcore hardware enjoys 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 optimisations, reviewing code changes, writing technical documentation, working with upstream communities, maintaining the code base, and supporting users.

You are expected to closely collaborate with other engineers, acting as a role model for junior members in your own team and working together with other teams, including those developing low-level libraries and ML applications on top of the frameworks.

The Team

The frameworks team is responsible for ensuring Graphcore hardware enjoys seamless support in user-facing machine learning frameworks such as Pytorch, Triton, Jax, and TensorFlow. We have a strong focus on both ease-of-use and performance.

Our software team is at the cutting edge of the machine learning industry, with a culture of driving continuous improvement and a deliberate focus on personal development. All our software teams follow agile working practices.

Responsibilities and Duties

• Designing and implementing new features & maintaining and supporting existing features.
• Developing and maintaining unit tests, component tests, and integration tests.
• Optimising the software stack to enhance performance for our users.
• Managing code quality, code complexity, and technical debt.
• 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.

Candidate Profile

Essential:

• Experience of developing performant C++ applications in a commercial setting.
• Experience of Python development.

Desirable:

• Experience using and/or development of ML Frameworks such as PyTorch.
• Demonstrable knowledge of AI/ML.
• Experience in a computationally intensive engineering field.
• Understanding of and/or experience with compiler development.
• Understanding of computing architectures.
• Experience of maintaining and supporting a complex code base.
• Experience of writing ML kernels.
• Experience in profiling/optimising high performance code.

Benefits

In addition to a competitive salary, Graphcore offers flexible working, a generous annual leave policy, private medical insurance and health cash plan, a dental plan, pension (matched up to 5%), life assurance, and income protection. We have a generous parental leave policy and an employee assistance programme (including health, mental wellbeing, and bereavement support). We offer a range of healthy food and snacks at our central Bristol office and have our own barista bar! We welcome people of different backgrounds and experiences; we’re committed to building an inclusive work environment that makes Graphcore a great home for everyone. We offer an equal opportunity process and understand that there are visible and invisible differences in all of us. We can provide a flexible approach to interviews and encourage you to chat with us if you require any reasonable adjustments.

Applicants for this position must hold the right to work in the UK. Unfortunately, at this time, we are unable to provide visa sponsorship or support for visa applications.