Google JAX Jobs in London

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

AI Research Engineer

London, UK
Hybrid / WFH Options
Arondite
language understanding The ability to write clean and maintainable code in Python, Rust, C++ or Java Experience with frameworks such as PyTorch, Tensorflow or JAX Strong written and verbal communication skills to document your work effectively and present it to relevant stake-holders and users A passion for identifying More ❯
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:

Software Engineer- AI/ML, AWS Neuron Distributed Training

London, United Kingdom
Amazon
control management, build processes, testing, and operations experience - Bachelor's degree in computer science or equivalent - Preferred previous software engineer expertise with Pytorch/Jax/Tensorflow, Distributed libraries and Frameworks, End-to-end Model Training. The group presents a lot of opportunity for optimization and scaling large deep 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:

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:

Member of Technical Staff, ML Performance

London, UK
Odyssey
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 - 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, UK

London, UK
graphcore
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 AI over More ❯
Posted:

Machine Learning Researcher

London, UK
Foster + Partners
Join to apply for the Machine Learning Researcher role at Foster + Partners . Foster + Partners is a global studio for architecture, engineering, urban and landscape design, rooted in sustainability. This role will include strategic R&D around developing More ❯
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:

Machine Learning Engineer London, UK

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:

Staff Software Engineer, GPU Performance, Core ML

London, United Kingdom
Google Inc
Staff Software Engineer, GPU Performance, Core ML corporate_fare Google place London, UK Apply Minimum Qualifications: Bachelor's degree or equivalent practical experience. 8 years of experience in software development and with data structures/algorithms (e.g., C++ or Python). 5 years of experience with Machine Learning (ML … 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:
Google JAX
London
25th Percentile
£86,525
Median
£88,863
75th Percentile
£91,202