10 of 10 Remote Google JAX Jobs in the UK

Science, Machine Learning, Mathematics, or a closely related field. Strong foundational knowledge of machine learning and deep learning , with hands-on experience using frameworks such as PyTorch , TensorFlow , or JAX . Exposure to LLM fine-tuning and/or experience working with transformer-based models (e.g. GPT, LLaMA, Mistral). Interest in or familiarity with agentic AI frameworks and More ❯

learn, etc.). Comfort working with large and complex datasets, including data wrangling and exploratory analysis. Experience (or strong interest) in GPU-powered model development (e.g. using PyTorch, TensorFlow, JAX, or similar). Demonstrated ability to work independently on real-world projects. Strong communicator, with the ability to share ideas, take feedback, and collaborate effectively. A master’s or More ❯

learn, etc.). Comfort working with large and complex datasets, including data wrangling and exploratory analysis. Experience (or strong interest) in GPU-powered model development (e.g. using PyTorch, TensorFlow, JAX, or similar). Demonstrated ability to work independently on real-world projects. Strong communicator, with the ability to share ideas, take feedback, and collaborate effectively. A master’s or More ❯

at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … or engineering roles. Strong expertise in Machine Learning and Deep Learning, with exposure to Reinforcement Learning as a plus. Proficiency in Python and modern ML libraries (e.g., TensorFlow, PyTorch, JAX, or Keras). Experience with version control systems (GitHub, GitLab) and knowledge of clean, maintainable coding practices. Familiarity with CI/CD pipelines for automating ML workflows. Ability to More ❯

may be a good fit if you: Have 5+ years of industry-related experience Are proficient in Python and have experience with deep learning frameworks such as PyTorch or Jax Have a strong software engineering background and are interested in working closely with researchers and other engineers Enjoy pair programming (we love to pair!) Care about code quality, testing … file types: pdf, doc, docx, txt, rtf Enter manually Accepted file types: pdf, doc, docx, txt, rtf (Optional) Personal Preferences * How do you pronounce your name? Website Publications (e.g. Google Scholar) URL Other Useful URLs (e.g. Blog Posts) Are you open to working in-person in one of our offices 25% of the time? * Select... Of the locations listed More ❯

e.g., NeurIPS, ICML, ICLR, ACL,EMNLP, NAACL, ICLR) with specific focus on agent systems, tool use, or multi-agentcoordination Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ) Excellent communication skills to report and present research findings anddevelopments clearly, both orally and in writing. Curious and innovative disposition capable of devising novel, well-founded algorithmicsolutions to … popular open-source libraries or writing production code. Experience training large-scale models over distributed nodes with cloud tools and providers such as Amazon AWS, MS Azure, LambdaLabs or Google Cloud. You will enjoy: Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting-edge methods and technologies. Plenty of data More ❯

Research Engineer, Machine Learning (Horizons) London, UK About Anthropic Anthropics mission is to create reliable, interpretable, and steerable AI systems. We want AI to be safe and beneficial for our users and for society as a whole. Our team is More ❯

The world will be unrecognisable in 5 years. Machine learning models are driving our cars , testing our eyesight , detecting our cancer , giving sight to the blind , giving speech to the mute , and dictating what we consume, enjoy, and think . More ❯

The world will be unrecognisable in 5 years. Machine learning models are driving our cars , testing our eyesight , detecting our cancer , giving sight to the blind , giving speech to the mute , and dictating what we consume, enjoy, and think . More ❯

at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … tools, and hardware optimisation techniques Experience with distributed training frameworks (Ray, Dask, PyTorch Lightning, etc.) Expertise with Python and/or C/C++ Development with machine learning frameworks (JAX, Tensorflow, PyTorch etc.) Passion for profiling, identifying bottlenecks, and delivering efficient solutions. Highly Desirable Track record of successfully scaling ML models. Experience writing custom CUDA kernels or XLA operations. More ❯