Google JAX Jobs in London

1 to 25 of 33 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:

Software Engineer (GenAI)

London, United Kingdom
Barlowe LLP
or a related field, or equivalent professional experience Strong programming skills in Python, with experience in machine learning frameworks such as PyTorch, TensorFlow or JAX Hands-on experience fine-tuning LLMs and optimising hyperparameters for improved performance Skilled in building LLM-powered applications for real-time decision-making and More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

AI Research Scientist (AEC)

London, United Kingdom
Merantix
Familiarity in statistical methods for Machine Learning (e.g. Bayesian methods, HMMs, graphical models, dimension reduction, clustering, classification, regression techniques, etc.) Familiarity with PyTorch, TensorFlow, JAX or similar frameworks Strong coding abilities in Python Preferred Qualifications Experience in the Architecture, Engineering, and/or Construction domains, including expertise with industry More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

AI Research Scientist: AEC

London, United Kingdom
Autodesk, Inc
Familiarity in statistical methods for Machine Learning (e.g. Bayesian methods, HMMs, graphical models, dimension reduction, clustering, classification, regression techniques, etc.) Familiarity with PyTorch, TensorFlow, JAX or similar frameworks Strong coding abilities in Python Preferred Qualifications Experience in the Architecture, Engineering, and/or Construction domains, including expertise with industry More ❯
Employment Type: Permanent
Salary: GBP Annual
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:

Research Scientist, FAIR (PhD)

London, UK
Meta
and algorithm scaling, or neural network architectures. Experience communicating complex research for public audiences of peers. Experience with deep learning frameworks such as Pytorch, Jax, or Tensorflow. Must obtain work authorization in country of employment at the time of hire, and maintain ongoing work authorization during employment. Preferred Qualifications More ❯
Posted:

ML Compiler Engineer, Annapurna Labs

London, UK
Amazon
software co-design - Experience with LLVM and/or MLIR - Experience with developing algorithms for simulation tools - Experience in TensorFlow, PyTorch, and/or JAX - Experience in LLM, Vision or other deep-learning models Amazon is committed to a diverse and inclusive workplace. Amazon is an equal opportunity employer More ❯
Posted:

Presales System Engineer - AI/HPC - (UK Based EMEA Role)

London, UK
Pure Storage
industry, including knowledge of the landscape and Pure’s solutions with a focus on the following: Deep understanding of industry AI frameworks (pytorch, tensorflow, jax etc) Deep understanding on high performance, low latency, highly parallelized compute solutions at scale (compute, network, storage) Deep understanding of accelerated compute environments (GPU More ❯
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:

Member of Technical Staff, ML Performance

London, UK
TN United Kingdom
of owning projects end to end Problem-solving mindset with the ability to acquire new skills as needed Proficiency with PyTorch (or TF/JAX), as well as NVIDIA GPU ecosystems and optimization stacks Highly metric-based Strong Python and C++ skills Bonus Qualifications Experience optimizing kernels with Triton More ❯
Posted:

Principal Solutions Architect, Generative AI, Startups, AWS

London, United Kingdom
ENGINEERINGUK
and deploying GenAI models and applications on cloud infrastructure. Sound knowledge of machine learning fundamentals, with working knowledge of frameworks such as Pytorch, TensorFlow, JAX or MXNet. Knowledge of MLOps tools and workflows for model development, validation, and deployment. High level of communicating effectively across internal and external organizations More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Solutions Architect, Generative AI, Startups, AWS

London, UK
Amazon
and deploying GenAI models and applications on cloud infrastructure. - Sound knowledge of machine learning fundamentals, with working knowledge of frameworks such as Pytorch, TensorFlow, JAX or MXNet. - Knowledge of MLOps tools and workflows for model development, validation, and deployment. - High level of communicating effectively across internal and external organizations More ❯
Posted:

Principal Solutions Architect, Generative AI, Startups, AWS

London, United Kingdom
Amazon
and deploying GenAI models and applications on cloud infrastructure. - Sound knowledge of machine learning fundamentals, with working knowledge of frameworks such as Pytorch, TensorFlow, JAX or MXNet. - Knowledge of MLOps tools and workflows for model development, validation, and deployment. - High level of communicating effectively across internal and external organizations More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Scientist

London, UK
Scindo
desire to work on the next generation of bioinformatics and biocatalysis predictive tools. Background in Python and experience with PyTorch; TensorFlow and/or JAX is a bonus, as well as knowledge of other programming languages Data preparation: feature selection, engineering, and data cleaning ML algorithm development and knowledge More ❯
Posted:

Machine Learning Engineer- World-Leading Prop Trading Fund - Oxford Knight

London, UK
Jobs via eFinancialCareers
techniques, model architecture, optimizer, learning rate schedules, etc. Thorough understanding of Python tools and libraries, keen to offer advice on best practices Experience using Jax, Tensorflow or similar ML frameworks a huge plus Benefits: Market-leading salaries Generous benefits package, including physical & mental health benefits, excellent holiday entitlement, significant More ❯
Posted:

Machine Learning Engineer- World-Leading Prop Trading Fund | London, UK

London, UK
Oxford Knight
techniques, model architecture, optimizer, learning rate schedules, etc. Thorough understanding of Python tools and libraries, keen to offer advice on best practices Experience using Jax, Tensorflow or similar ML frameworks a huge plus Benefits: Market-leading salaries Generous benefits package, including physical & mental health benefits, excellent holiday entitlement, significant More ❯
Posted:

Staff AI Research Engineer (Foundation Labs)

London, UK
Anterior
You've worked with open-source models like Llama/Deepseek You know how to get into the nuts and bolts of PyTorch/JAX You have experience in applied research with a product focus Preferred but Not Required You have experience building large-scale online evaluation pipelines You More ❯
Posted:

Research Scientist

London, United Kingdom
Graphcore
Job Summary As a researcher at Graphcore, you will contribute to the advancement of AI research, investigating new ideas that push the limits on important AI/ML problems. Specialised hardware has been the key driver of the progress of More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Engineer

London, UK
Graphcore
As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯
Posted:

ML Research Engineer

London, UK
Selby Jennings
Get AI-powered advice on this job and more exclusive features. Key Responsibilities: Develop and implement state-of-the-art machine learning models for trading and investment strategies. Research and apply techniques such as deep learning, reinforcement learning, and NLP More ❯
Posted:

Research Engineer, Foundational Research

London, UK
The Rundown AI, Inc
About Us Artificial Intelligence could be one of humanity’s most useful inventions. At Google DeepMind, we’re a team of scientists, engineers, machine learning experts and more, working together to advance the state of the art in artificial intelligence. We use our technologies for widespread public benefit and … develop and run experiments, helping to bring innovative ideas to reality. For this role a low-level understanding of common XLA primitives and how jax code runs on TPUs in practice is highly beneficial. The ideal candidate is willing to roll up sleeves to work across the LLM stack … flexibility and adaptability to work on any project is a must. In order to set you up for success as a Research Engineer at Google DeepMind, we look for the following skills and experience: BSc degree in computer science or a related STEM field Proven knowledge and experience in More ❯
Posted:

Machine Learning Engineer, Gemini Safety

London, UK
The Rundown AI, Inc
delivering state-of-the art language models to our users. About us Artificial Intelligence could be one of humanity’s most useful inventions. At Google DeepMind, we’re a team of scientists, engineers, machine learning experts and more, working together to advance the state of the art in artificial … of Safety and Alignment on LLMs. About you In order to set you up for success as a machine learning-focused software engineer at Google DeepMind, we look for the following skills and experience: MSc degree in computer science or machine learning or similar experience working in industry. 3+ … modeling and generative AI, working with data, model evaluations, and quality iterations. Proven knowledge and experience of Python, one or several machine learning frameworks (Jax, TensorFlow, or PyTorch), as well as algorithms and data structures. Ability to grow in high-stakes, high-visibility environments. In addition, the following would 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:
Google JAX
London
25th Percentile
£86,525
Median
£88,863
75th Percentile
£91,202