Remote Google JAX Jobs in London

9 of 9 Remote Google JAX Jobs in London

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:

Machine Learning Software Engineer, Research

London, UK
Hybrid / WFH Options
PhysicsX Ltd
scaler platforms, e.g., AWS, Azure, GCP); 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; C/C++ for computer vision, geometry processing, or scientific computing; software engineering concepts and best practices (e.g., versioning 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:

Quantitative Analyst - Sports Modelling

London, United Kingdom
Hybrid / WFH Options
Amelco
Bayesian inference, state-space models, filtering algorithms, etc. Excellent programming skills and experience implementing efficient algorithms for high-dimensional problems (ideally with TensorFlow, Julia, Google JAX). Any prior exposure to sports modelling would be a major advantage, but we will consider candidates with limited work experience. Why More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Machine Learning Engineer

London Area, United Kingdom
Hybrid / WFH Options
Adamas Knight
models that enhance data-driven drug discovery, precision medicine, and computational biology. Create and optimize multi-modal machine learning algorithms, using frameworks such as JAX, PyTorch, and TensorFlow, with applications to vision and language understanding. Collaborate with interdisciplinary teams, including computational biologists, bioinformaticians, and data scientists, to apply machine More ❯
Posted:

Manager, Research Science (Foundational Research, Machine Learning)

London, UK
Hybrid / WFH Options
TRSS
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. What’s in it For You? Join us to inform the way forward with the latest AI solutions and address real-world challenges More ❯
Posted:

Machine Learning Engineer in Genomics

London, UK
Hybrid / WFH Options
NLP PEOPLE
With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Lagos, 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 More ❯
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:

Senior Machine Learning Engineer - Team Lead

London, UK
Hybrid / WFH Options
InstaDeep
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 … and deploying large-scale, production-ready AI solutions, particularly in computer vision. Deep technical proficiency in machine learning, deep learning frameworks (e.g., PyTorch, TensorFlow, JAX), state-of-the-art models and architectures, as well as high-performance computing environments. Exceptional leadership and strategic vision, with experience guiding AI/ More ❯
Posted:
Google JAX
London
25th Percentile
£86,525
Median
£88,863
75th Percentile
£91,202