Machine Learning Engineer - PyTorch & JAX 6 Month+ Contract (Inside IR35) Rate - TBC Hybrid - 2 days - Cambridge office

Our multinational technology client is looking for a Machine Learning Engineer, with working experience using PyTorch & JAX. The ideal candidate will have strong experience working from PyTorch and JAX libraries. You will need to have exposure to distributed systems, coding and machine learning knowledge from a variety of businesses.

Key Responsibilities:

• Enhance coding experience, for machine learning deployments, using PyTorch & JAX.
  
  
• Collaborate with researchers to improve test and stability pipelines for large-scale machine learning.
  
  
• Optimise distributed systems for machine learning model deployment.
  
  
• Implement and refine scaling strategies for large-scale generative models.
  
  Experience Required:
  
  
• 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 work-flows.
  
  Everybody is welcome
  
  Diversity and Inclusion Statement. | PCR Digital
  
  "At PCR Digital, we are committed to ensuring that diversity, equity and inclusion play a role at all stages of our recruitment - it is important to us that our own company culture and the culture of our network is as varied and supportive as possible. We love people (it's why we do what we do), so, regardless of background, we welcome you to work with us or apply to any of our jobs if you feel that they are right for you."
  
  We also aim to ensure that our entire process is accessible. Please make us aware of any adjustments you may need throughout the selection, interview and general process and we will do all we can to ensure that any barriers are removed for you