4 of 4 Permanent Google JAX Jobs in Central London

applied AI research. Commercial or research experience implementing machine learning algorithms. Strong programming skills in Python; C++ experience is advantageous. Experience with PyTorch, TensorFlow, JAX, or similar deep learning frameworks. Why Apply? Join a world-class AI research team working on cutting-edge machine learning technologies. Play ...

Advanced proficiency in Python and C++ within a Linux/UNIX environment. Technical Stack: Extensive practical experience with leading machine learning frameworks (e.g., PyTorch , JAX , or TensorFlow ). Research Experience: A demonstrated track record of designing and implementing complex machine learning architectures to solve rigorous, data-heavy problems. Note ...

Lead AI Engineer Early employee at a high-growth AI startup Ideal Experience Strong Python software engineering skills Expertise with PyTorch and/or JAX Experience deploying machine learning systems into production Deep knowledge of LLMs and Generative AI Hands-on experience with LoRA, QLoRA, DPO, RLHF ...

Software Engineer III, GenAI Data Operations Research, XR corporate_fare Google place London, UK Bachelor’s degree or equivalent practical experience. 2 years of experience with software development in one or more programming languages, or 1 year of experience with an advanced degree. 1 year of experience with … enable end-to-end fine-tuning for foundational GenAI models and optimize them for production environments, including specialized hardware like TPUs. About the job Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. ...