PyTorch Frameworks Engineer — ML Systems & Performance

Cerebras; Cambridge; On-site; GBP 50,000 - 75,000; Full time

Job summary

A leading AI compute company in Cambridge is seeking software engineers to help design and support machine learning frameworks like PyTorch. The role offers a competitive salary, flexible working, and a supportive team environment. Successful candidates will have strong software engineering skills and experience in Python and C++. Join a diverse team committed to innovation and excellence in AI technology.

Benefits

Flexible working

Generous annual leave policy

Private medical insurance

Pension matched up to 5%

Employee assistance programme

Qualifications

• Demonstrable strong software engineering skills.
• Experience of Python development.
• Experience developing performant C++ applications in a commercial setting.

Responsibilities

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

Skills

Strong software engineering skills

Python development

C++ applications

Writing ML kernels

Compiler development

ML Frameworks development

Computationally intensive engineering

Job description

A leading AI compute company in Cambridge is seeking software engineers to help design and support machine learning frameworks like PyTorch. The role offers a competitive salary, flexible working, and a supportive team environment. Successful candidates will have strong software engineering skills and experience in Python and C++. Join a diverse team committed to innovation and excellence in AI technology.