Google JAX Job Vacancies

51 to 61 of 61 Google JAX Jobs

Machine Learning Scientist II/Sr (Biomedical Images) - UK

Bristol, Gloucestershire, United Kingdom
Iambic Therapeutics, Inc
supervised pre training, Vision Transformers, or vector quantization methods. Hands on work with large medical or scientific imaging datasets and establishing evaluation tasks and metrics. Proficiency in PyTorch or JAX and experience running distributed training on GPUs or cloud HPC resources. Strong software engineering habits: version control, testing, continuous integration, code review, and containerization. Clear communication and ability to More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Bio AI Research Scientist

London, United Kingdom
Prima Mente
ideally including transformers, diffusion models, or multimodal architectures. Experience applying ML to large multimodal biological datasets (genomics, transcriptomics, epigenomics, proteomics, single-cell data). Exceptional coding skills (Python, PyTorch, JAX, TensorFlow) and familiarity with scalable training frameworks. Experience managing large-scale distributed training across GPUs/TPUs. Background in generative AI, large language models, mechanistic interpretability, or reinforcement learning. More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

AI Research Scientist

London, United Kingdom
Antler
ideally including transformers, diffusion models, or multimodal architectures. Experience applying ML to large multimodal biological datasets (genomics, transcriptomics, epigenomics, proteomics, single-cell data). Exceptional coding skills (Python, PyTorch, JAX, TensorFlow) and familiarity with scalable training frameworks. Experience managing large-scale distributed training across GPUs/TPUs. Background in generative AI, large language models, mechanistic interpretability, or reinforcement learning. More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Principal Solutions Architect, Generative AI, Startups, AWS

London, United Kingdom
Amazon
professional. - Experience with building, validating 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, with both technical and non More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Machine Learning Engineer

Leuven, Vlaams-Brabant, Belgium
Hybrid / WFH Options
IC Resources
such as quantisation, pruning, and distillation to improve performance. Contribute to deployment, testing, and continuous model improvement. Requirements: Strong experience building and deploying ML models using PyTorch or TensorFlow; JAX is a plus. Solid background in NLP, computer vision and/or speech processing. Knowledge of model optimisation and performance tuning. Familiarity with distributed or edge AI environments. Passion More ❯
Employment Type: Permanent
Salary: EUR Annual
Posted:

Machine Learning Applied Researcher

London Area, United Kingdom
Harnham
Experience across 3D graphics, rendering, or materials/textures is required Experience implementing and training diffusion or generative image models Strong Python skills and familiarity with frameworks like PyTorch, JAX, or TensorFlow Understanding of transfer learning, computer vision, and image processing techniques Ability to translate complex research into scalable, production-ready solutions Why join this team Work with a More ❯
Posted:

Machine Learning Applied Researcher

City of London, London, United Kingdom
Harnham
Experience across 3D graphics, rendering, or materials/textures is required Experience implementing and training diffusion or generative image models Strong Python skills and familiarity with frameworks like PyTorch, JAX, or TensorFlow Understanding of transfer learning, computer vision, and image processing techniques Ability to translate complex research into scalable, production-ready solutions Why join this team Work with a More ❯
Posted:

25.05.02 Lead AI Scientist, Small Molecule Discovery, Orion Pharma, Cambridge, UK

Cambridge, Cambridgeshire, United Kingdom
Computational Chemistry List, Ltd. (CCL)
modern AI/ML methods (e.g., generative models, GNNs, active learning, foundation models)in a scientific context.- Strong coding skills and experience with relevant ML frameworks(e.g., PyTorch, TensorFlow, JAX) and chemistry toolkits (e.g., RDKit).- Deep understanding of molecular representation techniques and their application in modeling.- Experience in designing and implementing robust, scalable AI-driven workflows and tools. More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Applied Scientist

London, United Kingdom
Hybrid / WFH Options
UiPath
Life at UiPath The people at UiPath believe in the transformative power of automation to change how the world works. We're committed to creating category-leading enterprise software that unleashes that power. To make that happen, we need people More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

AI Scientist - Paris/London

London, United Kingdom
Mistral AI
proficient software engineer in a least one programming language (Python or other e.g. Rust, Go, Java) • You have, hands-on experience with AI frameworks and tools (e.g., TensorFlow, PyTorch, Jax) • You have high technical engineering competence. This means being able to design complex software and make it usable in production • You have a high scientific track record in a More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Principal Machine Learning Engineer / Team Lead

Oxford, England, United Kingdom
Hybrid / WFH Options
IC Resources
years of ML experience, with a strong record of transitioning research into production. Proven team leadership in a technical or research setting. Expert in Python and PyTorch; JAX is a plus. Strong foundation in data preparation, feature engineering, and model evaluation. Excellent communicator with the ability to adapt messaging for different audiences. Strategic mindset and leadership potential at the More ❯
Posted:
Google JAX
10th Percentile
£61,000
25th Percentile
£62,750
Median
£72,500
75th Percentile
£91,875
90th Percentile
£97,500