51 to 75 of 139 Google JAX Jobs in the UK excluding London

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 ❯

field • 5+ years of experience deploying AI/ML models in production settings • Proficient in Python, C++, and deep learning frameworks like PyTorch or Jax • Skilled in distributed training (e.g., DDP, FSDP) and model performance optimization • Experience with GPU architectures, cloud platforms, and hardware tradeoffs • Familiarity with low-level More ❯

Proven experience scaling cloud-native systems (AWS, microservices, IaC) Kubernetes + Terraform expertise in production environments Solid understanding of ML deployment (e.g. PyTorch, TensorFlow, JAX) Frontend skills with React/TypeScript or similar Bonus: big data/real-time analytics/multi-cloud Why Join? Equity with serious upside More ❯

. Hands-on experience training and deploying learned policies for complex behaviors in robots or simulated characters. Fluency with modern ML toolchains (PyTorch/JAX, RL or behavior-cloning libraries, agentic frameworks, MCP, distributed training workflows). Solid software-engineering fundamentals - tested, documented, maintainable Python & C++. Demonstrated ability to More ❯

LLM-based products (GPT, Llama, Mistral, or similar) or conversational AI systems into production. Deep knowledge of modern ML/NLP frameworks (PyTorch, TensorFlow, JAX) and orchestration stacks (LangChain, LlamaIndex, vector databases). Proven track record driving AI governance, model risk management, and compliance within regulated sectors (healthcare, med More ❯

of the latest trends in AI/ML research and development. Good familiarity with one of the popular ML frameworks such as PyTorch, TensorFlow, JAX, ONNX, or PaddlePaddle. A history of developer engagement or participation in an active open-source development community. Experience in Product Owner roles, or a More ❯

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 ❯

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 ❯

the analysis or application of data in finance, economics, or related fields is a plus. • Experience with machine learning libraries such as PyTorch, TensorFlow, JAX is a plus. • Experience applying generative AI solutions (via open source tools or through Azure, Amazon Bedrock, Nvidia etc.) is a plus. • Experience with … cloud technologies (Azure, AWS, Google Cloud Platform) is a plus. Our benefits To help you stay energized, engaged and inspired, we offer a wide range of employee benefits including: retirement investment and tools designed to help you in building a sound financial future; access to education reimbursement; comprehensive resources More ❯

Social network you want to login/join with: col-narrow-left Client: Founding Teams Location: Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 Posted: 16.06.2025 Expiry Date: 31.07.2025 col-wide Job Description: Founding More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, crawley, west sussex col-narrow-left Client: Founding Teams Location: crawley, west sussex, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, glasgow col-narrow-left Client: Founding Teams Location: glasgow, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, woking col-narrow-left Client: Founding Teams Location: woking, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, basildon col-narrow-left Client: Founding Teams Location: basildon, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, shrewsbury col-narrow-left Client: Founding Teams Location: shrewsbury, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, oxford district col-narrow-left Client: Founding Teams Location: oxford district, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, edinburgh col-narrow-left Client: Founding Teams Location: edinburgh, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, southampton col-narrow-left Client: Founding Teams Location: southampton, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

Social network you want to login/join with: col-narrow-left Client: Founding Teams Location: Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 Posted: 16.06.2025 Expiry Date: 31.07.2025 col-wide Job Description: Founding More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, worcester col-narrow-left Client: Founding Teams Location: worcester, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, newport col-narrow-left Client: Founding Teams Location: newport, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, bedford col-narrow-left Client: Founding Teams Location: bedford, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, norwich col-narrow-left Client: Founding Teams Location: Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 Posted: 16.06.2025 Expiry More ❯

Social network you want to login/join with: col-narrow-left Client: Founding Teams Location: Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 Posted: 16.06.2025 Expiry Date: 31.07.2025 col-wide Job Description: Founding More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, chesterfield col-narrow-left Client: Founding Teams Location: chesterfield, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯