3 of 3 Permanent Google JAX Jobs in the UK excluding London

scale. Proven track record of leading ML engineering projects from architecture to deployment, including ownership of production-grade systems. Deep expertise in ML frameworks and engineering stacks (TensorFlow, PyTorch, JAX, Ray, MLflow, Kubeflow). Proficiency in Python and at least one backend language (e.g., Java, Scala, Go, C++). Strong understanding of cloud ML infrastructure (AWS SageMaker, GCP Vertex More ❯

The best solution runs as much as possible on-device to enable low-power, accurate, and private AI products. This is where Plumerai leads, demonstrating better accuracy even than Google Nest. We are now starting to invest heavily in market adoption. We build the most accurate and efficient AI solutions by vertically integrating all layers of the stack. From … revenue is growing rapidly. We are backed by world-class investors such as Tony Fadell (creator of iPod, iPhone; founder of Nest), Hermann Hauser (founder of Arm), Zoubin Ghahramani (Google DeepMind), and others. Learn more here: TechCrunch: Tony Fadell-backed Plumerai brings on-device AI to home security cameras Plumerai raises $8.7M Series A to connect Vision LLMs to More ❯

About Mistral At Mistral AI, we believe in the power of AI to simplify tasks, save time, and enhance learning and creativity. Our technology is designed to integrate seamlessly into daily working life. We democratize AI through high-performance, optimized More ❯