Google JAX Jobs in the UK excluding London

17 of 17 Google JAX Jobs in the UK excluding London

AI Research Engineer

london, south east england, united kingdom
Oho Group Ltd
years of Relevant Experience building and applying ML systems Proficiency in Python, Rust, C++, or Java Experience with ML frameworks like PyTorch, TensorFlow, or JAX Excellent communication skills—written, verbal, and visual AI Research Engineer - Well-Funded Start-Up More ❯
Posted:

Machine Learning Scientist

Peterborough, Cambridgeshire, United Kingdom
Hyper Recruitment Solutions Ltd
machine learning, including deep learning and/or generative models. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Scientist

Cambridgeshire, England, United Kingdom
Hyper Recruitment Solutions
learning, including deep learning and/or generative models. 3. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling More ❯
Posted:

Machine Learning Scientist

cambridgeshire, east anglia, United Kingdom
Hyper Recruitment Solutions
learning, including deep learning and/or generative models. 3. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling More ❯
Posted:

Research Scientist | Generative Models | Foundational Models | Diffusion | LLMs |Biology

london, south east england, United Kingdom
Enigma
conferences. Expertise in Programming: Strong proficiency in software development and programming in Python. Experience with Deep Learning Frameworks: Proficiency in frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent Communication Skills: Ability to clearly communicate complex ideas, both verbally and in writing, and to work effectively within a More ❯
Posted:

Research Scientist

Peterborough, Cambridgeshire, United Kingdom
microTECH Global Limited
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 ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Engineer (Visual Specialist)

london, south east england, United Kingdom
Hybrid / WFH Options
OhChat
focus on GenAI 🚀 Strong expertise in Stable Diffusion, Flux, and similar architectures 🚀 Experience fine-tuning and optimising visual models 🚀 Hands-on experience with PyTorch, JAX, or TensorFlow 🚀 Proficiency in GPU inference optimisation (CUDA, BitsAndBytes, PEFT) 🚀 Experience deploying ML models using Kubernetes and/or Docker 🚀 Strong experience with Python More ❯
Posted:

Head of Applied AI

london, south east england, United Kingdom
Hybrid / WFH Options
Acquired Talent Ltd
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 ❯
Posted:

Lead AI Scientist - Small Molecule Discovery

Cambridge, Cambridgeshire, United Kingdom
Orion Pharma
e.g., generative models, GNNs, active learning, foundation models) in a scientific context. Strong coding skills and experience with relevant ML frameworks (e.g., PyTorch, TensorFlow, JAX) and chemistry toolkits (e.g., RDKit). Deep understanding of molecular representation techniques and their application in modeling. Experience in designing and implementing robust, scalable More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

25.05.02 Lead AI Scientist, Small Molecule Discovery, Orion Pharma, Cambridge, UK

Cambridge, Cambridgeshire, United Kingdom
The Computational Chemistry List
e.g., generative models, GNNs, active learning, foundation models)in a scientific context.- Strong coding skills and experience with relevant ML frameworks(e.g., PyTorch, TensorFlow, JAX) and chemistry toolkits (e.g., RDKit).- Deep understanding of molecular representation techniques and their application in modeling.- Experience in designing and implementing robust, scalable More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior AI Engineer

Cambridge, England, United Kingdom
DeepRec.ai
Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and More ❯
Posted:

Senior AI Engineer

cambridge, east anglia, United Kingdom
DeepRec.ai
Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and More ❯
Posted:

Senior AI Engineer

Cambridge, south west england, United Kingdom
DeepRec.ai
Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and More ❯
Posted:

Machine Learning Researcher - LLM/VLM

Surrey, United Kingdom
Redline Group Ltd
Machine Learning Researcher - LLM/VLM Are you a PhD-educated Machine Learning Researcher looking for a new opportunity? If so, our client, a global consumer electronics company, is actively expanding their team. This role is based at one of More ❯
Employment Type: Permanent
Salary: GBP 45,000 - 70,000 Annual
Posted:

Machine Learning Researcher - LLM/VLM

Cambridge, Cambridgeshire, Staines, Surrey, United Kingdom
Redline Group Ltd
Machine Learning Researcher - LLM/VLM Are you a PhD-educated Machine Learning Researcher looking for a new opportunity? If so, our client, a global consumer electronics company, is actively expanding their team. This role is based at one of More ❯
Employment Type: Permanent
Salary: £45000 - £70000/annum
Posted:

AI Research Engineer

london, south east england, United Kingdom
Understanding Recruitment
AI Research Engineer (London) A fast-growing defence tech startup focused on AI and software for autonomous systems is hiring an AI Research Engineer for their remote team. Partnered with a third-party recruiter, this well-funded company is committed More ❯
Posted:

Machine Learning Engineer

london, south east england, United Kingdom
GCS AI
Machine Learning Engineer | Research Platform: We are a tier 1 supplier for a well-known buy-side quant firm on the lookout for a Machine Learning Engineer (£475,000-£515,000) who is happy to work at the interface of More ❯
Posted:
Google JAX
the UK excluding London
10th Percentile
£62,300
25th Percentile
£62,750
Median
£67,000
75th Percentile
£72,750
90th Percentile
£74,100