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.