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