Machine Learning Research Engineer (PyTorch/JAX) – Cambridge

Research Machine Learning Engineer (PyTorch/Jax) – Technology consumer – Cambridge

Day rate: £500 – £700 (inside IR35)

Duration: 12 months

Start: ASAP

My new client is looking for a Research Machine Learning Engineer to join the team on a contract basis in their Cambridge office. The ideal candidate would have Pytorch and Jax experience.

Job Description

Responsibilities:

• Enhance codebase for high-performance machine learning deployments using PyTorch
• Collaborate with researchers to improve test and stability pipelines for large-scale machine learning
• Optimize distributed systems for machine learning model deployment
• Implement and refine scaling strategies for large-scale generative models

Requirements:

• Strong proficiency in PyTorch and distributed systems
• Deep understanding of scaling challenges and benefits for large-scale machine learning models
• Expertise in generative modelling techniques
• Experience with test automation and stability pipelines for ML systems
• Ability to collaborate effectively with research teams
• Strong problem-solving and debugging skills

Preferred Qualifications:

• Advanced degree in Computer Science, Machine Learning, or related field
• Published research or significant contributions to open-source ML projects
• Experience with large distributed ML workflows.