Google JAX Job Vacancies

26 to 50 of 74 Google JAX Jobs

Research Scientist

Peterborough, Cambridgeshire, United Kingdom
microTECH Global Limited
experience in academia or industry • We will consider various levels of experience in relevant research areas Key Skills: • Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) • Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Scientist

Peterborough, Cambridgeshire, United Kingdom
Hyper Recruitment Solutions Ltd
machine learning, including deep learning and/or generative models. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Scientist

Cambridgeshire, England, United Kingdom
Hyper Recruitment Solutions
learning, including deep learning and/or generative models. 3. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling More ❯
Posted:

Machine Learning Scientist

cambridgeshire, east anglia, united kingdom
Hyper Recruitment Solutions
learning, including deep learning and/or generative models. 3. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling More ❯
Posted:

Senior AI Engineer

United States
Ada Meher
across the stack, from architecture to deployment What they're looking for: Strong experience with Python and deep learning frameworks like PyTorch, TensorFlow, or JAX Familiarity with libraries and tooling such as Transformers (Hugging Face), TorchVision, OpenCV, scikit-learn, and Weights & Biases (or similar for experiment tracking) Skilled in More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior AI Engineer

San Francisco, California, United States
Ada Meher
across the stack, from architecture to deployment What they're looking for: Strong experience with Python and deep learning frameworks like PyTorch, TensorFlow, or JAX Familiarity with libraries and tooling such as Transformers (Hugging Face), TorchVision, OpenCV, scikit-learn, and Weights & Biases (or similar for experiment tracking) Skilled in More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior AI Engineer

San Jose, California, United States
Ada Meher
across the stack, from architecture to deployment What they're looking for: Strong experience with Python and deep learning frameworks like PyTorch, TensorFlow, or JAX Familiarity with libraries and tooling such as Transformers (Hugging Face), TorchVision, OpenCV, scikit-learn, and Weights & Biases (or similar for experiment tracking) Skilled in More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior AI Engineer

Sonoma, California, United States
Ada Meher
across the stack, from architecture to deployment What they're looking for: Strong experience with Python and deep learning frameworks like PyTorch, TensorFlow, or JAX Familiarity with libraries and tooling such as Transformers (Hugging Face), TorchVision, OpenCV, scikit-learn, and Weights & Biases (or similar for experiment tracking) Skilled in More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior AI Engineer

San Mateo, California, United States
Ada Meher
across the stack, from architecture to deployment What they're looking for: Strong experience with Python and deep learning frameworks like PyTorch, TensorFlow, or JAX Familiarity with libraries and tooling such as Transformers (Hugging Face), TorchVision, OpenCV, scikit-learn, and Weights & Biases (or similar for experiment tracking) Skilled in More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior AI Engineer

Hayward, California, United States
Ada Meher
across the stack, from architecture to deployment What they're looking for: Strong experience with Python and deep learning frameworks like PyTorch, TensorFlow, or JAX Familiarity with libraries and tooling such as Transformers (Hugging Face), TorchVision, OpenCV, scikit-learn, and Weights & Biases (or similar for experiment tracking) Skilled in More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior AI Engineer

Alameda, California, United States
Ada Meher
across the stack, from architecture to deployment What they're looking for: Strong experience with Python and deep learning frameworks like PyTorch, TensorFlow, or JAX Familiarity with libraries and tooling such as Transformers (Hugging Face), TorchVision, OpenCV, scikit-learn, and Weights & Biases (or similar for experiment tracking) Skilled in More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior AI Engineer

Sunnyvale, California, United States
Ada Meher
across the stack, from architecture to deployment What they're looking for: Strong experience with Python and deep learning frameworks like PyTorch, TensorFlow, or JAX Familiarity with libraries and tooling such as Transformers (Hugging Face), TorchVision, OpenCV, scikit-learn, and Weights & Biases (or similar for experiment tracking) Skilled in More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior AI Engineer

Fremont, California, United States
Ada Meher
across the stack, from architecture to deployment What they're looking for: Strong experience with Python and deep learning frameworks like PyTorch, TensorFlow, or JAX Familiarity with libraries and tooling such as Transformers (Hugging Face), TorchVision, OpenCV, scikit-learn, and Weights & Biases (or similar for experiment tracking) Skilled in More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior AI Engineer

Santa Rosa, California, United States
Ada Meher
across the stack, from architecture to deployment What they're looking for: Strong experience with Python and deep learning frameworks like PyTorch, TensorFlow, or JAX Familiarity with libraries and tooling such as Transformers (Hugging Face), TorchVision, OpenCV, scikit-learn, and Weights & Biases (or similar for experiment tracking) Skilled in More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior AI Engineer

Santa Clara, California, United States
Ada Meher
across the stack, from architecture to deployment What they're looking for: Strong experience with Python and deep learning frameworks like PyTorch, TensorFlow, or JAX Familiarity with libraries and tooling such as Transformers (Hugging Face), TorchVision, OpenCV, scikit-learn, and Weights & Biases (or similar for experiment tracking) Skilled in More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Machine Learning Engineer

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

Artificial Intelligence Engineer

United Kingdom
NineTech
Key Tech the client are using: 🧠 LLMs (GPT, Claude, LLaMA, etc.) 🛠️ LangChain, AutoGPT, CrewAI 🤖 Reinforcement Learning & Multi-Agent Systems 📡 Kafka, RabbitMQ, OpenTelemetry 🧰 TensorFlow, PyTorch, JAX ☁️ AWS, Azure, Vertex AI What You'll Do: • Build and deploy autonomous agents • Optimize real-time AI models • Implement RL and planning algorithms • Integrate 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:

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:

Lead AI Scientist - Small Molecule Discovery

Cambridge, Cambridgeshire, United Kingdom
Orion Pharma
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 More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

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

Cambridge, Cambridgeshire, United Kingdom
The Computational Chemistry List
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 More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior AI Engineer

Cambridge, England, United Kingdom
DeepRec.ai
Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and More ❯
Posted:

Senior AI Engineer

cambridge, east anglia, united kingdom
DeepRec.ai
Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and More ❯
Posted:
Google JAX
10th Percentile
£62,900
25th Percentile
£64,250
Median
£79,432
75th Percentile
£91,202