Frameworks Engineer: PyTorch & Triton Performance

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

Job summary

A technology company in the UK is seeking a Framework Engineer to design and support software for machine learning frameworks like PyTorch. The role involves optimizing code performance, developing features, and mentoring junior engineers. Candidates must have experience in C++ and Python development. Benefits include flexible working, private medical insurance, and a generous leave policy. Applicants must have the right to work in the UK without requiring visa sponsorship.

Benefits

Flexible working

Generous annual leave

Private medical insurance

Health cash plan

Dental plan

Pension (matched up to 5%)

Life assurance

Income protection

Parental leave policy

Qualifications

• Must have experience developing performant C++ applications in a commercial setting.
• Experience with Python development is essential.
• Desirable to have experience using ML frameworks like PyTorch.

Responsibilities

• Design and implement new features and maintain existing ones.
• Develop and maintain unit, component, and integration tests.
• Optimize software stack for performance.
• Manage code quality and technical debt.
• Conduct code reviews and documentation.
• Coach and mentor team members.

Skills

Experience of developing performant C++ applications

Experience of Python development

Tools

PyTorch

Job description

A technology company in the UK is seeking a Framework Engineer to design and support software for machine learning frameworks like PyTorch. The role involves optimizing code performance, developing features, and mentoring junior engineers. Candidates must have experience in C++ and Python development. Benefits include flexible working, private medical insurance, and a generous leave policy. Applicants must have the right to work in the UK without requiring visa sponsorship.