Remote Google JAX Jobs in the UK

8 of 8 Remote Google JAX Jobs in the UK

Machine Learning Engineer

London, United Kingdom
Hybrid / WFH Options
Locus Robotics
MAS). Advanced Python programming skills, with a strong emphasis on writing efficient, scalable, and maintainable code. Proven experience with TensorFlow/PyTorch/Jax, Scikit-learn, and MLOps workflows for training, deployment, and monitoring of ML models. Experience working with Polars and/or Pandas for high-performance More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Head of Data Science

London Area, United Kingdom
Hybrid / WFH Options
KDR Talent Solutions
advanced machine learning techniques, with a focus on LLMs/genAI, optimisation, or sequential learning Strong background in Python, deep learning frameworks (TensorFlow, PyTorch, Jax), and ML model development PhD in Computer Science, Machine Learning, Applied Mathematics, or related field, or equivalent experience A history of successful scientific publications More ❯
Posted:

Head of Data Science

london, south east england, united kingdom
Hybrid / WFH Options
KDR Talent Solutions
advanced machine learning techniques, with a focus on LLMs/genAI, optimisation, or sequential learning Strong background in Python, deep learning frameworks (TensorFlow, PyTorch, Jax), and ML model development PhD in Computer Science, Machine Learning, Applied Mathematics, or related field, or equivalent experience A history of successful scientific publications More ❯
Posted:

Research Data Scientist

London, United Kingdom
Hybrid / WFH Options
PhysicsX Ltd
data-driven role, with exposure to: building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., NumPy, SciPy, Pandas, PyTorch, JAX), especially including deep learning applications; developing models for bespoke problem settings that involve high-dimensional data (spatiotemporal, geometric, physical); iterating on network architectures and More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Head of Applied AI

London Area, 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:

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:

Manager, Research Science (Foundational Research, Machine Learning)

London, United Kingdom
Hybrid / WFH Options
Thomas Reuters
record in top-tier conferences (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP, NAACL, ICLR). Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ) Experience in ML Research beyond completing a PhD (e.g. supervision, industry experience, leading academic initiatives, ). Excellent communication skills to report and present … source libraries or writing production code. Experience training large-scale models over distributed nodes with cloud tools such as Amazon AWS, MS Azure, or Google Cloud. Other key points: Plenty of data, compute, and high-impact problems: Our scientists and engineers get to explore large datasets and discover new More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Research Scientist, Generative AI (Multimodal Foundation Models)

London, United Kingdom
Hybrid / WFH Options
InstaDeep Ltd
With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select … by publications in reputable scientific journals or conferences. Expertise in software development and programming in Python. Experience with deep learning frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent ability to communicate complex ideas clearly, both verbally and in writing, and to work effectively within a multidisciplinary team. More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:
Google JAX
10th Percentile
£62,900
25th Percentile
£64,250
Median
£79,432
75th Percentile
£91,202