Frameworks Engineer: PyTorch & Triton Performance

Cerebras; Cambridge; On-site; GBP 70,000 - 110,000; Full time

Benefits offered by this job

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

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.

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.