13 of 13 Remote Google JAX Jobs in the UK excluding London

e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible More ❯

frameworks (e.g., PyTorch, TensorFlow)Solid grasp of ML techniques; independent and team-oriented mindset Preferred : Experience in LLMs, reinforcement learning, or chip designFamiliarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applicationsCollaborate with top researchers in a global teamFlexible hybrid work More ❯

e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindset Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible More ❯

experience in academia or industry We will consider various levels of experience in relevant research areas Key Skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU More ❯

experience in academia or industry We will consider various levels of experience in relevant research areas Key Skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU More ❯

. Hands-on experience training and deploying learned policies for complex behaviors in robots or simulated characters. Fluency with modern ML toolchains (PyTorch/JAX, RL or behavior-cloning libraries, agentic frameworks, MCP, distributed training workflows). Solid software-engineering fundamentals - tested, documented, maintainable Python & C++. Demonstrated ability to More ❯

LLM-based products (GPT, Llama, Mistral, or similar) or conversational AI systems into production. Deep knowledge of modern ML/NLP frameworks (PyTorch, TensorFlow, JAX) and orchestration stacks (LangChain, LlamaIndex, vector databases). Proven track record driving AI governance, model risk management, and compliance within regulated sectors (healthcare, med More ❯

and Qualifications MS or PhD in CS/EE or equivalent experience in the industry, with key skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantisation, pruning, sparsification, etc.) Experience with deployment on embedded and mobile devices (ML inference and/or training) Experience with More ❯

the analysis or application of data in finance, economics, or related fields is a plus. • Experience with machine learning libraries such as PyTorch, TensorFlow, JAX is a plus. • Experience applying generative AI solutions (via open source tools or through Azure, Amazon Bedrock, Nvidia etc.) is a plus. • Experience with … cloud technologies (Azure, AWS, Google Cloud Platform) is a plus. Our benefits To help you stay energized, engaged and inspired, we offer a wide range of employee benefits including: retirement investment and tools designed to help you in building a sound financial future; access to education reimbursement; comprehensive resources More ❯

collaborating across functions, especially with engineering, hardware, and system design teams Solid programming and prototyping skills using modern deep learning frameworks (e.g. PyTorch, TensorFlow, JAX) Location: London (hybrid, with 4 days onsite and flexible hours) This is a high-impact role for someone who thrives on building ambitious technology More ❯

experience, with a strong record of transitioning research into production. Proven team leadership in a technical or research setting. Expert in Python and PyTorch; JAX is a plus. Strong foundation in data preparation, feature engineering, and model evaluation. Excellent communicator with the ability to adapt messaging for different audiences. More ❯

Experience shipping real products to customers ML Design and Deployment: Familiarity with designing, training, optimizing, and deploying machine learning models using frameworks like PyTorch, JAX, LiteRT, HuggingFace Transformers, or commercial APIs. Containerisation: Experience with cloud deployments and containerisation (Docker, Kubernetes, containerd). Linux: Embedded Linux and kernel development. Programming More ❯

Permanent Role Location : Cambridge Hybrid Policy: 3 days onsite and 2 days working from home weekly Role and Responsibilities As anAI Research Scientist, you will be part of a dynamic team working on next-generation machine learning technologies for real More ❯