Permanent Google JAX Jobs in London

1 to 25 of 30 Permanent Google JAX Jobs in London

Senior Data Scientist

Greater London, England, United Kingdom
GlobalLogic
in AI research and applied machine learning. Expertise in key data science technologies and techniques, such as Python, Git, AWS, AWS Sagemaker, PyTorch, TensorFlow, Jax, Numpy, Scikit-learn, Time-series forecasting, Classification, Regression, Large-language models, and Experiment Design. Strong stakeholder management skills and the ability to communicate complex More ❯
Posted:

Senior Data Scientist

london, south east england, United Kingdom
GlobalLogic
in AI research and applied machine learning. Expertise in key data science technologies and techniques, such as Python, Git, AWS, AWS Sagemaker, PyTorch, TensorFlow, Jax, Numpy, Scikit-learn, Time-series forecasting, Classification, Regression, Large-language models, and Experiment Design. Strong stakeholder management skills and the ability to communicate complex More ❯
Posted:

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:

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 Researcher Scientist

London, United Kingdom
Autodesk, Inc
regression techniques, etc.) Familiarity with Deep Learning techniques (e.g. Network architectures, regularization techniques, learning techniques, loss-functions, optimization strategies, etc.) Familiarity with PyTorch , TensorFlow , JAX or similar frameworks Strong coding abilities in Python and/or C++ Preferred Qualifications: 2D & 3D Generative AI Reinforcement Learning LLMs and Natural Language 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:

Staff Software Engineer, GPU Performance, Core ML

London, United Kingdom
Google
Staff Software Engineer, GPU Performance, Core ML link Copy link corporate_fare Google place London, UK Advanced Experience owning outcomes and decision making, solving ambiguous problems and influencing stakeholders;deep expertise in domain. Apply link Copy link Bachelor's degree or equivalent practical experience. 8 years of experience in … and performance bottlenecks. Ability to develop and utilize sophisticated performance models and benchmarks to guide optimization efforts and hardware roadmap decisions. About the job Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our … and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google's needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Deep Learning Researcher

london, south east england, United Kingdom
microTECH Global LTD
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 If interested please apply below or send your CV to Trei@microtech-global.com More ❯
Posted:

Deep Learning Researcher

west london, south east england, United Kingdom
microTECH Global LTD
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 If interested please apply below or send your CV to Trei@microtech-global.com More ❯
Posted:

Deep Learning Researcher

south west london, south east england, United Kingdom
microTECH Global LTD
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 If interested please apply below or send your CV to Trei@microtech-global.com More ❯
Posted:

AI/ML Engineer - Senior, Principal & Team Lead - Team Build

london, south east england, United Kingdom
Hybrid / WFH Options
Aventis Solutions
models at scale, ideally within product environments or a data-rich business. Tech-savvy : Proficient in Python and modern ML frameworks (e.g., PyTorch, TensorFlow, JAX). Strong command of cloud-native AI tooling (e.g., AWS SageMaker, GCP Vertex AI, AzureML/AzureAI, MLflow, Weights & Biases). Well-rounded Engineer More ❯
Posted:

Senior Machine Learning Engineer

London Area, United Kingdom
KDR Talent Solutions
Learning, Computer Science, or related field Proven track record with LLMs, generative AI, optimization, or sequential learning Experience with deep learning frameworks (TensorFlow, PyTorch, JAX, etc.) Expertise in one or more: reinforcement learning, Bayesian optimization, optimal experimental design, agents, or causality Experience in drug/target discovery, treatment modeling More ❯
Posted:

Senior Machine Learning Engineer

london, south east england, United Kingdom
KDR Talent Solutions
Learning, Computer Science, or related field Proven track record with LLMs, generative AI, optimization, or sequential learning Experience with deep learning frameworks (TensorFlow, PyTorch, JAX, etc.) Expertise in one or more: reinforcement learning, Bayesian optimization, optimal experimental design, agents, or causality Experience in drug/target discovery, treatment modeling More ❯
Posted:

Data Science Modeller

London, United Kingdom
Hybrid / WFH Options
Lloyds Banking Group
Strong Data Engineering underpinnings and an ability to work with big data (Tbs). Experience in at least one Deep Learning framework (PyTorch, TensorFlow, JAX etc). Strong leadership skills, using empathy, clarity and challenge to support your team to be their best. Strong communication skills with an ability More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Generative AI Model and Distributed Training, APJ

London, United Kingdom
Amazon
applications to engineering teams and C-Level executives. This requires deep familiarity across the stack - compute infrastructure (Amazon EC2, Amazon EKA), ML frameworks PyTorch, JAX, orchestration layers Kubernetes and Slurm, parallel computing (NCCL, MPI), MLOPs, through to Amazon SageMaker Hyperpod, Amazon Bedrock as well as target use cases in More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

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:

Senior/ Staff Research Engineer

London Area, United Kingdom
Hybrid / WFH Options
GCS
field • 5+ years of experience deploying AI/ML models in production settings • Proficient in Python, C++, and deep learning frameworks like PyTorch or Jax • Skilled in distributed training (e.g., DDP, FSDP) and model performance optimization • Experience with GPU architectures, cloud platforms, and hardware tradeoffs • Familiarity with low-level More ❯
Posted:

Senior/ Staff Research Engineer

london, south east england, United Kingdom
Hybrid / WFH Options
GCS
field • 5+ years of experience deploying AI/ML models in production settings • Proficient in Python, C++, and deep learning frameworks like PyTorch or Jax • Skilled in distributed training (e.g., DDP, FSDP) and model performance optimization • Experience with GPU architectures, cloud platforms, and hardware tradeoffs • Familiarity with low-level More ❯
Posted:

AI Intern

London Area, United Kingdom
Hybrid / WFH Options
Capua
from a top-tier university. • Solid experience in training neural networks, tuning hyperparameters, and working with frameworks like PyTorch. • Knowledge of differentiable programming (e.g., JAX) or geometric deep learning (e.g., PyTorch Geometric) is a plus. • Strong foundation in mathematical concepts such as stochastic calculus, optimisation, and simulation techniques. • Proficiency More ❯
Posted:

AI Intern

london, south east england, United Kingdom
Hybrid / WFH Options
Capua
from a top-tier university. • Solid experience in training neural networks, tuning hyperparameters, and working with frameworks like PyTorch. • Knowledge of differentiable programming (e.g., JAX) or geometric deep learning (e.g., PyTorch Geometric) is a plus. • Strong foundation in mathematical concepts such as stochastic calculus, optimisation, and simulation techniques. • Proficiency More ❯
Posted:

Machine Learning Engineer

London, United Kingdom
Woven by Toyota
to-end stack; Experience MS, or higher degree, in a related field, or equivalent industry experience Professional experience with ML frameworks such as PyTorch, Jax or Tensorflow (PyTorch preferred) Experience with data sampling and data curation pipelines for autonomous driving datasets Experience in state-of-the-art architectures for More ❯
Employment Type: Permanent
Salary: GBP Annual
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:

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:

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: