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 will 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.