8 of 8 Google JAX Jobs

Machine Learning & Coding Expert (SME)

Hiring Organisation
Meridial Marketplace, by Invisible
Location
United Kingdom
Explainable AI (XAI) SHAP/LIME interpretation, fairness flagging Data Storytelling Generating readable summaries from charts or model outputs Tools TensorFlow/PyTorch/JAX/HuggingFace We offer a pay range of $25-to- $100 per hour, with the exact rate determined after evaluating your experience, expertise ...

Machine Learning Quant Engineer

Hiring Organisation
Michael Page
Location
City Of London, England, United Kingdom
This temporary role requires an ML Quant Engineer with expertise within an Investment Bank. The position is based in London and involves developing and implementing machine learning models to support financial decision-making. Client Details ...

Quantitative Developer (Python) - Hybrid London - Up To 250k

Hiring Organisation
Opus Recruitment Solutions
Location
London, United Kingdom
Employment Type
Permanent
Salary
GBP 150,000 - 250,000 Annual
candidates who: Have deep familiarity with Python data ecosystem Understanding of Jupyter notebooks Exposure to machine learning libraries like PyTorch, XGBoost and JAX Understanding of crypto or traditional financial markets Strong API design and documentation skills What do you get in return? Up to 250k base (depending on experience ...

Quantitative Developer Python - Hybrid London - Up To 250k!

Hiring Organisation
Opus Recruitment Solutions Ltd
Location
Central London, London, England, United Kingdom
Employment Type
Full-Time
Salary
£150,000 - £250,000 per annum
candidates who: Have deep familiarity with Python data ecosystem Understanding of Jupyter notebooks Exposure to machine learning libraries like PyTorch, XGBoost and JAX Understanding of crypto or traditional financial markets Strong API design and documentation skills What do you get in return? Up to £250k base (depending on experience ...

Founding Machine Learning Engineer

Hiring Organisation
Bjak
Location
Etzelwang, Bayern, Germany
Employment Type
Permanent
Salary
EUR 80,000 - 110,000 Annual
Strong background in deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills - writing robust, production-grade systems Experience with … Experience training or deploying multimodal or diffusion models Experience in large-scale data processing (Apache Arrow, Spark, Ray) Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI) What You'll Get Extreme ownership and autonomy from day one - you define and build key model systems. Founding ...

Senior Software Engineer, AI Fitness, Health Coach, Fitbit

Hiring Organisation
Google
Location
London, England, United Kingdom
maintaining, or launching software products, and 1 year of experience with software design and architecture. Experience using machine learning frameworks (e.g., Tensorflow, PyTorch, or Jax). Experience with machine learning model pretraining, fine-tuning, evaluation, and methodologies. Preferred qualifications: Master's degree or PhD in Computer Science … related technical field. Experience publishing in high-quality machine learning journals. Experience with Python and C++. 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 products need to handle information ...

Machine Learning Consultant

Hiring Organisation
Lynx Recruitment Ltd
Location
Melbourn, Royston, Hertfordshire, England, United Kingdom
Employment Type
Full-Time
Salary
Salary negotiable
*Must be Sole British National for this role* This is an exciting opportunity to join a cutting-edge technology consultancy working at the forefront of innovation in defence and homeland security. As a Machine Learning ...

Senior ML Systems Engineer, Frameworks & Tooling

Hiring Organisation
Cohere
Location
London, England, United Kingdom
runs. You Might Be a Good Fit If You Have Strong engineering experience in large-scale distributed training or HPC systems. Deep familiarity with JAX internals, distributed training libraries, or custom kernels/fused ops. Experience with multi-node cluster orchestration (Slurm, Ray, Kubernetes, or similar). Comfort debugging … closely with infra, research, and deployment teams. Nice to Have Experience with training LLMs or other large transformer architectures. Contributions to ML frameworks (PyTorch, JAX, DeepSpeed, Megatron, xFormers, etc.). Familiarity with evaluation and serving frameworks (vLLM, TensorRT-LLM, custom KV caches). Experience with data pipeline optimization, sharded ...