76 to 100 of 117 Google JAX Jobs in the UK

Senior Engineering Manager – Machine Learning

Location
United Kingdom
Senior Engineering Manager - Machine Learning United Kingdom We are on a mission to pioneer the world’s next era of play. As we grow across Europe and Latin America, we’re building The Playstack - the ...

VLA Pre-Training Deep Learning Engineer

Location
Greater London, England, United Kingdom
data quality, and ensure continuous training cycles. Preferred qualifications include 3+ years in deep learning, proficiency in Python and frameworks like PyTorch/JAX, and effective communication. The role offers 23 days off, private healthcare, and a pension scheme. #J-18808-Ljbffr ...

Football Analytics Research Engineer — Hybrid

Location
London Colney, England, United Kingdom
from data to UI. Applicants should have strong quantitative skills and experience in Python/SQL/TypeScript and PyData stack, plus PyTorch/JAX/TF for model development. #J-18808-Ljbffr ...

ML Researcher for AI-Driven Trading Signals

Location
Greater London, England, United Kingdom
performance, with no prior finance requirement. Candidates hold a Master’s or PhD in ML, CS, or statistics, and excel in Python, Torch/JAX/TF, #J-18808-Ljbffr ...

Research Engineer

Location
London Colney, England, United Kingdom
live training/matchday schedule, not just be technically neat. Python/SQL/TypeScript, PyData stack (Pandas/Polars, Altair, etc.), PyTorch/JAX/TF. You'll own things end-to-end, data through UI. Looking for strong quant background, first-principles engineering judgment, and taste ...

Senior Staff Machine Learning Engineer, Content Platform

Location
Greater London, England, United Kingdom
have strong experience building production-grade machine learning systems at scale You are experienced with modern ML frameworks such as PyTorch, TensorFlow, or JAX You have worked with or are interested in multimodal machine learning You understand how to design systems that balance automation with quality, safety, and user ...

Technical Lead, Machine Learning

Location
Greater London, England, United Kingdom
work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable ...

Research Scientist, LLM Agents (Foundational Research)

Hiring Organisation
Thomson Reuters
Location
London, United Kingdom
Salary
£ 60 K
NAACL, ICLR) with specific focus on agent systems, tool use, or multi-agent coordination Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ...)Excellent communication skills to report and present research findings and developments clearly, both orally and in writing.Curious and innovative disposition capable … writing production code.Experience training large-scale models over distributed nodes with cloud tools and providers such as Amazon AWS, MS Azure, LambdaLabs or Google Cloud.You will enjoy:Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting-edge methods and technologies.Plenty ...

Research Scientist, LLM Agents (Foundational Research)

Location
Greater London, England, United Kingdom
NAACL, ICLR) with specific focus on agent systems, tool use, or multi-agent coordination. Familiarity with one or more deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ...). Excellent communication skills to report and present research findings and developments clearly, both orally and in writing. Curious and innovative disposition … writing production code. Experience training large-scale models over distributed nodes with cloud tools and providers such as Amazon AWS, Microsoft Azure, LambdaLabs or Google Cloud. You will enjoy Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting-edge methods ...

Manager, Lead Research Scientist, LLM Agents (Foundational Research)

Location
Greater London, England, United Kingdom
NAACL, ICLR) with specific focus on agent systems, tool use, or multi‐agent coordination. 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 … 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. You will enjoy Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting‐edge methods ...

Manager, Lead Research Scientist, Training Data (Foundational Research)

Location
Greater London, England, United Kingdom
EMNLP, NAACL, ICLR) ) with specific focus on training data curation, synthetic data generation, etc.* 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 … 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.*You will enjoy:** Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting-edge methods ...

Staff Systems Architect for AI Inference Platform

Location
Greater London, England, United Kingdom
Olix Computing seeks an Architect/Staff Systems Software Engineer to own the DX-1 runtime and serving stack, connecting PyTorch/JAX to the hardware and enabling production inference at rack scale. This is a whole-stack role at the intersection of runtime, network, and accelerator with close ...

Founding Research Engineer

Location
Greater London, England, United Kingdom
decisions Breadth across model architectures — transformers, diffusion, Mamba and others; the field hasn't converged and curiosity matters Fluency in Python and PyTorch or JAX Systems-level thinking: reasons from first principles, not from defaults Fully fluent in English — written and verbal, hard requirement Must be based in Europe ...

Director, Artificial Intelligence & Machine Learning Engineer

Location
City of Edinburgh, Scotland, United Kingdom
About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat’spossible in asset management. Today, AI is reshaping every aspect of our industry, from ...

Vice President, Artificial Intelligence & Machine Learning Engineer

Hiring Organisation
Hackajob Ltd
Location
Edinburgh, Midlothian, Scotland, United Kingdom
Employment Type
Permanent, Work From Home
hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

Software Engineer

Hiring Organisation
Experis
Location
London, United Kingdom
Employment Type
Contract
Contract Rate
£750 - £830/day
design and commitment to the development of reusable, scalable, and testable software components. Basic understanding of at least one major deep learning framework (PyTorch, JAX, TensorFlow). Hands-on experience with Google Cloud Platform, in particular the services we build on: Cloud Run, Google Kubernetes Engine, Cloud ...

Machine Learning Engineer, Applied AI - Deployed

Location
Greater London, England, United Kingdom
better outcomes, 80% less admin work Raised a $30M Series A from top investors Built a team of 40+ AI experts from Tesla, Google DeepMind, NVIDIA, and Databricks At Brain Co., your work will be deployed in the real world, not stuck in research. We move fast, with more … high level familiarity with the architecture and operation of large language models. Have personally implemented models in common ML frameworks such as PyTorch, Jax or TensorFlow. Possess a strong foundation in data structures, algorithms, and software engineering principles. Exhibit excellent problem‐solving and analytical skills, with a proactive approach ...

AI engineer (UK)

Location
Greater London, England, United Kingdom
About WRITER WRITER is where the world’s leading enterprises orchestrate AI-powered work. Our vision is to expand human capacity through superintelligence. And we’re proving it is possible – through powerful, trustworthy AI that ...

Director, Artificial Intelligence & Machine Learning Engineer

Hiring Organisation
Hackajob Ltd
Location
Edinburgh, Midlothian, Scotland, United Kingdom
Employment Type
Permanent, Work From Home
hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefine what's possible ...

Research Scientist, Gemini Omni, DeepMind

Location
Greater London, England, United Kingdom
learning or equivalent practical experience. Experience with training generative models (Large Language Model (LLM), image, video). Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience conducting research. Publication record in AI conferences (e.g., NeurIPS, CVPR, ICCV, ICLR). Preferred qualifications Experience with large-scale data … pipelines. Experience with training large-scale models. Proven experience conducting research in industry About the job As an organization, Google maintains a portfolio of research projects driven by fundamental research, new product innovation, product contribution and infrastructure goals, while providing individuals and teams the freedom to emphasize specific types ...

Research Scientist, Gemini Omni, DeepMind

Hiring Organisation
Google
Location
London, United Kingdom
Salary
£ 70 K
statistics, machine learning or equivalent practical experience.Experience with training generative models (Large Language Model (LLM), image, video).Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch).Experience conducting research.Publication record in AI conferences (e.g., NeurIPS, CVPR, ICCV, ICLR).Preferred qualifications:Experience with large-scale data pipelines.Experience with training … large-scale models.Proven experience conducting research in industryAs an organization, Google maintains a portfolio of research projects driven by fundamental research, new product innovation, product contribution and infrastructure goals, while providing individuals and teams the freedom to emphasize specific types of work. As a Research Scientist, you'll setup ...

Research Engineer, Gemini Omni, DeepMind

Location
Greater London, England, United Kingdom
Switzerland . Master's degree in computer science, mathematics, applied statistics, machine learning or equivalent practical experience. Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience working in industry, working on projects from proof-of-concept through to implementation. Experience with training diffusion models. Experience conducting … pipelines. Experience with training large-scale models. Cross-functional collaboration experience. Knowledge of machine learning, statistics, and diffusion model theories. About the job At Google, research-focused Software Engineers are embedded throughout the company, allowing them to setup large-scale tests and deploy promising ideas quickly and broadly. Ideas ...

Research Engineer, Gemini Omni, DeepMind

Location
Greater London, England, United Kingdom
Minimum qualifications: Master's degree in computer science, mathematics, applied statistics, machine learning or equivalent practical experience. Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch). Experience working in industry, working on projects from proof-of-concept through to implementation. Experience with training diffusion models. Experience conducting … pipelines. Experience with training large-scale models. Cross-functional collaboration experience. Knowledge of machine learning, statistics, and diffusion model theories. About the job At Google, research-focused Software Engineers are embedded throughout the company, allowing them to setup large-scale tests and deploy promising ideas quickly and broadly. Ideas ...

Research Engineer, Gemini Omni, DeepMind

Hiring Organisation
Google
Location
London, United Kingdom
Salary
£ 80 K
scale.Minimum qualifications:Master's degree in computer science, mathematics, applied statistics, machine learning or equivalent practical experience.Experience of TensorFlow or ML frameworks (e.g. JAX or PyTorch).Experience working in industry, working on projects from proof-of-concept through to implementation.Experience with training diffusion models.Experience conducting applied research.Preferred qualifications:Experience … inference optimization.Experience with data pipelines.Experience with training large-scale models.Cross-functional collaboration experience.Knowledge of machine learning, statistics, and diffusion model theories. At Google, research-focused Software Engineers are embedded throughout the company, allowing them to setup large-scale tests and deploy promising ideas quickly and broadly. Ideas may come ...