Graduate Software Engineer — ML Frameworks & Triton

Graphcore; West of England; On-site; GBP 32,000 - 46,000; Full time

Benefits offered by this job

Flexible working

Generous leave

Pension matched

Phantom equity

On-site barista

Job summary

Graphcore is seeking a Graduate Software Engineer to join the Triton team, building software that helps ML developers leverage Graphcore hardware. You will connect accelerators with popular ML frameworks and learn how compilers, frameworks and hardware interact in production settings.

You will contribute to features, tests, docs, code reviews and performance improvements while receiving mentorship from experienced engineers.

Qualifications

• Bachelor’s or master’s degree in computer science, Maths, Machine Learning, Data Science, or a related field.
• Hands-on experience with Python or C++, gained through study, projects, placements, internships or open source.
• Practical experience or strong academic understanding of compiler development; interest in AI, ML frameworks, high performance software or computing architectures.
• Ability to write clear, maintainable code, tests and documentation.
• Curiosity, ownership and willingness to learn through feedback, code reviews and technical discussion.

Responsibilities

• Build software that helps ML developers get more from Graphcore hardware.
• Contribute to features, tests, documentation, code reviews and performance improvements.
• Learn how compilers, frameworks and hardware interact in production engineering.

Skills

Python

C++

Code quality

Education

BSc/MSc in CS/Math/ML/Data Science

Tools

PyTorch

Triton

JAX

TensorFlow

Job description

Graphcore is seeking a Graduate Software Engineer to join the Triton team, building software that helps ML developers leverage Graphcore hardware. You will connect accelerators with popular ML frameworks and learn how compilers, frameworks and hardware interact in production settings.

You will contribute to features, tests, docs, code reviews and performance improvements while receiving mentorship from experienced engineers.