14 of 14 Remote Google JAX Jobs in London

Join to apply for the Research Engineer, Machine Learning role at Mistral AI Continue with Google Continue with Google Join to apply for the Research Engineer, Machine Learning role at Mistral AI 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 … our enterprise API About You Masters or PhD in 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 chops Strong software … Technology Industries Technology, Information and Internet Referrals increase your chances of interviewing at Mistral AI by 2x Sign in to set job alerts for Research Engineer roles. Continue with Google Continue with Google Continue with Google Continue with Google London, England, United Kingdom 3 weeks ago Hampton, England, United Kingdom 6 days ago Quantum Algorithms Scientist More ❯

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 ❯

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. … language processing and computer vision) in industry. Master, PhD degree or equivalent experience in applied mathematics, computer science, bioinformatics or related scientific fields. Proficiency in software engineering (Python, Pytorch, JAX, Docker, Linux). A willingness to learn and develop skills in computational genomics including building an understanding of biological concepts related to cancer. Excellent communication skills in English. Appropriate More ❯

NeurIPS, ICML, ICLR, ACL, EMNLP, NAACL, ICLR with specific focus on agent systems, tool use, or multi-agent coordination. Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ) Experience in ML Research beyond completing a PhD (e.g. supervision, industry experience, leading academic initiatives, ). Excellent communication skills to report and present research findings and developments clearly … code contributions to popular open-source libraries or writing production code. Experience training large-scale models over distributed nodes with cloud tools such as Amazon AWS, MS Azure, 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 ❯

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 ❯

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 ❯

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 ❯

Life at UiPath The people at UiPath believe in the transformative power of automation to change how the world works. We're committed to creating category-leading enterprise software that unleashes that power. To make that happen, we need people More ❯