1 to 25 of 33 Google JAX Jobs

Senior AI Engineer - Google AI & Generative Intelligence

Hiring Organisation
Eclaro
Location
Paramus, New Jersey, United States
Employment Type
Permanent
Salary
USD Annual
Senior AI Engineer - Google AI & Generative Intelligence Job Number: 26-00855 Ready for a rewarding opportunity in the Financial Services Industry? ECLARO is looking for a Senior AI Engineer - Google AI & Generative Intelligence for our client in Paramus, NJ. ECLARO's client is a market-leading insurance company … challenge, then take a chance at this rewarding opportunity! Position Overview: A highly experienced Senior AI Engineer with deep expertise in Google AI technologies, Generative AI. Brings 10-15 years of broad software engineering experience, with the last 4 years focused exclusively on Artificial Generative Intelligence, including designing, building ...

Machine Learning Engineer

Hiring Organisation
Jobleads-UK
Location
Greater London, England, United Kingdom
Strong software engineering skills: clean code, testing, debugging, version control, and system design. Proficiency in Python with experience in ML frameworks (PyTorch, TensorFlow, or JAX). Hands‐on experience with cloud platforms (AWS, GCP, or Azure) and containerisation (Docker, Kubernetes). Solid understanding of ML fundamentals (model training, evaluation ...

Senior AI Engineer

Hiring Organisation
Aubay UK
Location
London Area, United Kingdom
/ML solutions in industry or engineering environments. Strong expertise in Python and AI/ML libraries: PyTorch, TensorFlow, Scikit‐Learn, Keras, NumPy, Pandas, JAX, XGBoost, PyTorch Geometric, OpenCV, Transformers, LangChain, Pinecone . Proven hands‐on experience deploying models in production and building scalable AI software. Deep understanding ...

Senior Software Engineer

Hiring Organisation
Jobleads-UK
Location
Cambridge, England, United Kingdom
equivalent experience in Computer Science, Maths, Machine Learning, Data Science, or related field. Desirable: Exposure to machine learning frameworks such as PyTorch, JAX, Triton, TensorFlow Experience with distributed workload management systems such as Kubernetes, VLLM, Keras or MLOps pipelines Experience working with hardware simulators or emulators (e.g. QEMU). ...

Machine Learning Engineer

Hiring Organisation
Findrs
Location
Aylesbury, England, United Kingdom
probability, optimisation, statistics, and ML fundamentals • Strong Python programming skills, with C++ experience considered highly beneficial • Experience using frameworks such as PyTorch, TensorFlow, or JAX • Experience building scalable ML systems for production environments • Familiarity with cloud or GPU based training infrastructure and MLOps workflows • Experience designing evaluation, testing ...

Distinguished AI Engineer

Hiring Organisation
Jobleads-UK
Location
Greater London, England, United Kingdom
/ML incubations or complex deep‐tech cross‐functional initiatives with measurable outcomes. Expertise in Python and deep experience with deep‐learning frameworks (e.g., Jax, TensorFlow, PyTorch) and state‐of‐the‐art ML techniques. Practical experience with modern AI paradigms such as world models, foundation models, multi‐modal language ...

Lead ML Researcher

Hiring Organisation
microTECH Global LTD
Location
Cambridge, England, United Kingdom
learning, and/or large language models (LLMs) Experience in chip design and applied AI research Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience influencing organizational policy and securing major research funding ...

Principal ML Researcher

Hiring Organisation
Microtech Global Ltd
Location
Cambridge, Cambridgeshire, East Anglia, United Kingdom
Employment Type
Permanent
learning, and/or large language models (LLMs) Experience in chip design and applied AI research Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience influencing organizational policy and securing major research funding ...

Research Engineer, Machine Learning – Paris/London/Zurich/Warsaw

Hiring Organisation
Jobleads-UK
Location
Greater London, England, United Kingdom
Computer Science (or equivalent proven track record). 4 + years working on large‐scale ML codebases. Hands‐on with PyTorch, JAX or TensorFlow; comfortable with distributed training (DeepSpeed/FSDP/SLURM/K8s). Experience in deep learning, NLP or LLMs; bonus for CUDA or data‐pipeline ...

Machine Learning Engineer

Hiring Organisation
Krotos
Location
United Kingdom
awareness of latency and cost tradeoffs MLOps competency — experiment tracking, model versioning, evaluation pipelines, ML CI Experience with Python and modern ML frameworks (PyTorch, JAX) Excellent verbal and written English communication skills Excellent analytical and problem-solving skills A desire to innovate and push current practice Desirable: Experience with ...

Principal Research Scientist London, United Kingdom

Hiring Organisation
Jobleads-UK
Location
Greater London, England, United Kingdom
have been instrumental in: building machine learning models and pipelines in Python, using common libraries and frameworks (PyTorch/CUDA, ideally with exposure to JAX, NumPy/SciPy), especially including deep learning applications; developing models for bespoke problem settings that involve high‐dimensional data (spatiotemporal, geometric, physical); iterating ...

AI Scientist / Machine Learning Engineer

Hiring Organisation
Jobleads-UK
Location
Greater London, England, United Kingdom
machine learning models for time-series or sensor data Strong programming skills in Python Experience with machine learning frameworks such as PyTorch, TensorFlow, or JAX Experience building data pipelines and training workflows Ability to work with imperfect, noisy, or incomplete datasets Nice to Have Experience with physiological signal processing ...

Deep Learning Researcher

Hiring Organisation
microTECH Global LTD
Location
London Area, United Kingdom
abilities, including technical documentation Experience with optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience with scalable system design and deployment in research or production environments Exposure to chip design or applied AI research Track record of contributing ...

Machine Learning Engineer

Hiring Organisation
Jobleads-UK
Location
Greater London, England, United Kingdom
members in diverse fields Strong programming abilities in Python, and extensive experience with the scientific and machine-learning stack: Numpy, Torch/Tensorflow/Jax, Scikit-learn, Polars, SQL Expertise with deep learning approaches such as diffusion or flow matching Proficiency in modern software development practices: code testing, documentation ...

Head of Applied AI Research

Hiring Organisation
Higher - AI recruitment
Location
City of London, London, United Kingdom
reinforcement learning Experience with agent frameworks, tool-use planning and workflow orchestration Systems-level fluency across data pipelines, training infrastructure and inference (PyTorch, JAX, TensorFlow) Strong MLOps discipline: CI/CD for models, offline/online evaluation, safety and guardrails The ability to influence at exec level and mentor ...

Senior Computational Biologist | ML Engineer

Hiring Organisation
SEQUENTIAL
Location
Cambridge, England, United Kingdom
knowledge of ML evaluation, validation strategy, and failure modes including error analysis and iteration planning ● Experience with model development using PyTorch/TensorFlow/JAX ● Ability to communicate clearly across technical + scientific stakeholders ● Strong experience leading or coordinating cross-functional teams, and thinking strategically across product and science ...

Research scientist with focus on HPC, AI, and training (f/m/d)

Hiring Organisation
Universität Stuttgart - Höchstleistungsrechenzentrum
Location
Stuttgart, Baden-Württemberg, Germany
Employment Type
Permanent
Salary
EUR Annual
work in bwHPC-S5. Ideally, your profile is supplemented by: Experience with AI frameworks, workloads (e.g. Natural Language Processing), and libraries (e.g., PyTorch or JAX). Experience with parallel programming paradigms. Some background in the field of (green) transport (e.g., fleet optimisation, energy-efficient logistics, smart city planning). ...

Senior Computer Vision Engineer - Robotics

Hiring Organisation
Pro Contract Jobs Ltd
Location
London, United Kingdom
Employment Type
Contract
learning systems, with shipped models or published research. Hands-on experience with Vision Language Models, LLMs, or generative models. Strong Python and PyTorch/JAX skills, including debugging, profiling, and shipping. A passion for bridging the gap between simulation and reality. This is a technically challenging role with ...

Machine Learning Engineer

Hiring Organisation
Hyper Recruitment Solutions LTD
Location
London, King's Cross, United Kingdom
Employment Type
Permanent
3. A working knowledge and practical experience with Python, and extensive experience with the scientific and machine-learning stack: Numpy, Torch/Tensorflow/Jax, Scikit-learn, Polars, SQL. Key Words: Machine Learning Engineer/Data Processing/Model Building/NMR/Deep Learning/Computational Biophysics/ ...

Senior Machine Learning Engineer

Hiring Organisation
SR2 | Socially Responsible Recruitment | Certified B Corporation™
Location
City of London, London, United Kingdom
deep learning and modern foundation model architectures Experience with probabilistic modelling techniques such as Bayesian methods, Gaussian Processes, or variational inference Expertise in PyTorch, JAX, or similar frameworks Experience training large models on distributed infrastructure Strong software engineering and systems design capabilities Nice to Have Background in computational biology ...

Machine Learning Engineer- Reinforcement Learning

Hiring Organisation
Wave Recruitment
Location
City of London, London, United Kingdom
agents already running in production What We're Looking For Essential 3-5 years training and deploying deep RL agents in Python PyTorch or JAX, and RL libraries such as Gymnasium A background in physical systems - engineering (mechanical, electrical, structural, biomedical), physics, robotics, autonomous driving, or control systems ...

Principal Machine Learning Engineer

Hiring Organisation
Jobleads-UK
Location
United Kingdom
Hands‐on experience training, fine‐tuning, or deploying large‐scale ML models in production. Proficiency with at least one modern ML framework (e.g. PyTorch, JAX), and ability to learn others quickly. Experience with distributed training and inference frameworks (e.g. DeepSpeed, FSDP, Megatron, ZeRO, Ray). Strong software engineering fundamentals ...

Research Engineer

Hiring Organisation
Cubiq Recruitment
Location
City of London, London, United Kingdom
excited by applying those skills to one of the most interesting data domains in AI. Areas of interest include: PyTorch/JAX internals CUDA, XLA or Triton optimisation Distributed training systems GPU cluster performance tuning Large-scale experiment management Training reliability and reproducibility Kubernetes and orchestration tooling Infrastructure supporting ...

Staff Machine Learning Engineer

Hiring Organisation
Jobleads-UK
Location
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 ...

Machine Learning Research Engineer (Foundational Research)

Hiring Organisation
Jobleads-UK
Location
Greater London, England, United Kingdom
software engineering practices* Demonstrated expertise in building production-quality code and data pipelines for ML systems* Proficiency in modern AI development frameworks including: PyTorch, Jax , HuggingFace Transformers, LLM APIs (litellm etc) and vLLM for building and deploying large-scale AI applications* Understanding of LLM training methodologies including instruction fine … generation, data curation, or curriculum learning pipelines* Experience training large-scale models over distributed nodes with cloud tools such as AWS, MS Azure, or Google Cloud* Hands-on experience with MLOps, experiment tracking, and model deployment systems* Strong interest in staying current with ML research literature and ability ...