Research Machine Learning Engineer (PyTorch/Jax) - Technology consumer - CambridgeDay rate: 500 - 700 (inside IR35)Duration: 12 monthsStart: ASAPMy 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.ResponsibilitiesJob DescriptionEnhance codebase for high-performance machine learning deployments using PyTorchCollaborate with researchers to improve test and stability pipelines for large-scale machine learningOptimize distributed systems for machine learning model deploymentImplement and refine scaling strategies for large-scale generative modelsRequirementsStrong proficiency in PyTorch and distributed systemsDeep understanding of scaling challenges and benefits for large-scale machine learning modelsExpertise in generative modelling techniquesExperience with test automation and stability pipelines for ML systemsAbility to collaborate effectively with research teamsStrong problem-solving and debugging skillsPreferred QualificationsAdvanced degree in Computer Science, Machine Learning, or related fieldPublished research or significant contributions to open-source ML projectsExperience with large distributed ML workflows.











Show more















Show less