PyTorch Frameworks Engineer – Flexible Work & Performance

Cerebras; Cambridge; On-site; GBP 80,000 - 100,000; Full time

Job summary

A leading AI technology company is seeking a framework engineer to design and implement features in machine learning frameworks like PyTorch. The role involves performance optimization, code reviews, and collaboration with cross-functional teams. The ideal candidate has strong C++ and Python skills, alongside experience in machine learning applications. This position offers competitive benefits, including flexible working and personal development opportunities.

Benefits

Flexible working

Private medical insurance

Pension matched up to 5%

Life assurance

Generous leave policy

Qualifications

• Experience developing performant C++ applications in a commercial setting.
• Experience in Python development.
• Experience using and/or developing ML frameworks such as PyTorch is a plus.

Responsibilities

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

Skills

C++ application development

Python development

Machine Learning Frameworks

Performance optimization

Code review

Job description

A leading AI technology company is seeking a framework engineer to design and implement features in machine learning frameworks like PyTorch. The role involves performance optimization, code reviews, and collaboration with cross-functional teams. The ideal candidate has strong C++ and Python skills, alongside experience in machine learning applications. This position offers competitive benefits, including flexible working and personal development opportunities.