4 of 4 Google JAX Jobs in the UK

production ML systems. Deep ML expertise, with strong foundations in math, statistics, and computer science. Proficiency in modern ML frameworks (PyTorch/TensorFlow/JAX) and distributed systems. Exceptional communication skills—able to simplify complex topics for executives and provide precise technical guidance to teams. Values: Kind, motivated, self ...

software engineering practices Demonstrated expertise in building production-quality code and data pipelines for ML systems Proficiency in modern AI development frameworks including: PyTorch, Jax , HuggingFace Transformers, LLM APIs (litellm etc) and vLLM for building and deploying large-scale AI applications Understanding of LLM training methodologies including instruction fine … generation, data curation, or curriculum learning pipelines Experience training large-scale models over distributed nodes with cloud tools such as AWS, MS Azure, or Google Cloud Hands-on experience with MLOps, experiment tracking, and model deployment systems Strong interest in staying current with ML research literature and ability ...

software engineering practices Demonstrated expertise in building production-quality code and data pipelines for ML systems Proficiency in modern AI development frameworks including: PyTorch, Jax , HuggingFace Transformers, LLM APIs (litellm etc) and vLLM for building and deploying large-scale AI applications Understanding of LLM training methodologies including instruction fine … generation, data curation, or curriculum learning pipelines Experience training large-scale models over distributed nodes with cloud tools such as AWS, MS Azure, or Google Cloud Hands-on experience with MLOps, experiment tracking, and model deployment systems Strong interest in staying current with ML research literature and ability ...

*Must be Sole British National for this role* This is an exciting opportunity to join a cutting-edge technology consultancy working at the forefront of innovation in defence and homeland security. As a Machine Learning ...