Google JAX Jobs in England

15 of 15 Google JAX Jobs in England

AI Solutions Engineer

London, United Kingdom
Bioptimus
deep experience with the AWS ecosystem and services relevant to ML, such as SageMaker, EC2, S3, IAM, and VPC. You are familiar with deep-learning frameworks like PyTorch or JAX, and libraries from the AI ecosystem like Hugging Face Transformers. You have worked with the lifecycle of AI models, including deployment, hosting, and optimization for low-latency inference. You More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

AI & Machine Learning Engineer, Associate

London, United Kingdom
Hybrid / WFH Options
LGBT Great
environments. Proficiency in production-grade programming languages such as Rust and C++. Decent understanding of distributed systems, cluster orchestration and management. Good knowledge of data science tools (e.g PyTorch, Jax, Numpy) and programming languages such as Python. Experience with monitoring tools (Prometheus, Grafana). Experience working in Agile development teams with excellent collaboration skills. Grit in the face of More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Sr. Machine Learning Engineer, Amazon General Intelligence (AGI)

London, United Kingdom
Amazon
languages such as Python, C++ or Rust; hands-on experience with parallel computing platforms such as CUDA, OpenMP, etc.; deep understanding of AI frameworks such as PyTorch, TensorFlow, and JAX, and their demands on underlying compute infrastructure, memory bandwidth, network interconnect, and storage as scale goes up; knowledge of emerging AI hardware accelerators and architectures; experience with containerization and More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

AI Engineer

London, United Kingdom
Deepstreamtech
record of building and deploying LLM-powered applications in production , Deep experience applying large language models to complex, real-world problems , Strong proficiency in Python and ML frameworks (PyTorch, JAX, TensorFlow, Hugging Face Transformers) , Hands-on experience with advanced LLM techniques: RAG systems, chain-of-thought prompting, agentic tool use, and multi-agent coordination , Experience with distributed systems, cloud More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Research Scientist, AI for Human Simulation

London, United Kingdom
Autodesk, Inc
learning and generative AI techniques, including implementing custom architectures, pretraining/fine-tuning generative models, building agentic workflows, and deploying solutions with real-world constraints. Familiarity with PyTorch, TensorFlow, JAX or similar frameworks Preferred Qualifications Experience in building AI agents (based on RL, LLMs, etc.) Experience in building multi-agent simulation Background in Large Language Models/NLP and More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior ML Frameworks Engineer

London, United Kingdom
Grid Dynamics International, Inc
with existing data and machine learning systems Qualifications Strong understanding of machine learning principles, especially in the context of LLMs. Solid proficiency in Python, including machine learning packages like Jax/Tensorflow or PyTorch Experience with large-scale data infrastructure Strong verbal and written communication skills with the ability to work effectively across internal and external organisations and virtual More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Applied Scientist Applied Science

London, United Kingdom
Hybrid / WFH Options
Unlikely Artificial Intelligence Limited
drive: Deep learning expertise with 2+ years in language generation. Experience with LLMs, neurosymbolic integration, and knowledge representation. Strong Python skills and experience with ML frameworks (PyTorch, HF, TensorFlow, JAX). Demonstrable track record of end-to-end applied research. Excellent communication skills; proven mentoring ability. Collaborative and proactive in fast-paced environments. Desirable: Experience with symbolic reasoning engines. More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Software Engineer - AI Systems & Infrastructure

London, England, United Kingdom
Harnham
tooling for deployment, monitoring, and continuous improvement of AI models What We're Looking For: Strong programming skills (Python, C++, or similar) Experience with ML frameworks (e.g. PyTorch, TensorFlow, JAX) Familiarity with cloud infrastructure and distributed systems Solid understanding of performance optimization and system design More ❯
Posted:

Software Engineer - AI Systems & Infrastructure

slough, south east england, united kingdom
Harnham
tooling for deployment, monitoring, and continuous improvement of AI models What We're Looking For: Strong programming skills (Python, C++, or similar) Experience with ML frameworks (e.g. PyTorch, TensorFlow, JAX) Familiarity with cloud infrastructure and distributed systems Solid understanding of performance optimization and system design More ❯
Posted:

Software Engineer - AI Systems & Infrastructure

london, south east england, united kingdom
Harnham
tooling for deployment, monitoring, and continuous improvement of AI models What We're Looking For: Strong programming skills (Python, C++, or similar) Experience with ML frameworks (e.g. PyTorch, TensorFlow, JAX) Familiarity with cloud infrastructure and distributed systems Solid understanding of performance optimization and system design More ❯
Posted:

Research Engineer - ML

London, England, United Kingdom
Harnham
to refine training workflows Contribute to the development of robust baselines and scalable pipelines What We're Looking For: Strong Python skills and experience with ML frameworks (e.g., PyTorch, JAX) Familiarity with model fine-tuning and training evaluation at scale Ability to translate research into practical implementations Background in generative models and/or 3D data is a plus More ❯
Posted:

Research Scientist

London, United Kingdom
Cerebras
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:

Research Scientist, Generative AI (Multimodal Foundation Models)

London, United Kingdom
Hybrid / WFH Options
InstaDeep Ltd
at the forefront of innovation. 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 NVIDIA Elite Service Delivery Partner. … of high-quality research, evidenced 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. Strong analytical and critical thinking More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Applied Scientist

London, United Kingdom
Unlikely Artificial Intelligence Limited
At UnlikelyAI, we're looking for a Senior Applied Scientist to join our Applied Science team. This is a high-impact individual contributor position. You'll help drive the end-to-end lifecycle of projects: from identifying opportunities in literature More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Staff Applied Scientist Applied Science

London, United Kingdom
Hybrid / WFH Options
Unlikely Artificial Intelligence Limited
Overview At UnlikelyAI, we are building the future of AI: one that is reliable, accurate and transparent. Our neurosymbolic technology harnesses the power of LLMs and generative AI, and combines it with classical symbolic technology to produce hallucination-resistant artificial More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:
Google JAX
England
10th Percentile
£66,000
25th Percentile
£75,000
Median
£95,000
75th Percentile
£97,500