51 to 59 of 59 Google JAX Jobs in the UK

We are supporting a pioneering technology organisation seeking a Senior Research Engineer to lead complex AI research and engineering initiatives. This position sits at the intersection of advanced research and practical engineering. You will help ...

Software Engineer III, GenAI Data Operations Research, XRGoogleLondon, United Kingdom**Description**Software Engineer III, GenAI Data Operations Research, XR \_corporate\_fare\_ Google \_place\_ London, UK \*\*Mid\*\* Experience driving progress, solving problems, and mentoring more junior team members; deeper expertise and applied knowledge within relevant area. \*\*Minimum qualifications:\*\* + Bachelor … enable end-to-end fine-tuning for foundational GenAI models and optimize them for production environments, including specialized hardware like TPUs. \*\*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. ...

supervised fine-tuning (SFT), or agent-based systems, including modeling or applied infrastructure. 2 years of experience with machine learning frameworks such as JAX, Flax, TensorFlow, or PyTorch, and scaling experiments. Preferred qualifications PhD in Computer Science, Machine Learning, or a related quantitative field. Publications in reinforcement learning … artifacts. Preference for simple, elegant solutions to research problems. Experience with systems design and managing code complexity in large codebases. About the job At Google DeepMind, we've built a unique culture and work environment where long-term ambitious research can flourish. In this role, you will push ...

Software Engineer III, GenAI Data Operations Research, XR corporate_fare Google place London, UK Bachelor’s degree or equivalent practical experience. 2 years of experience with software development in one or more programming languages, or 1 year of experience with an advanced degree. 1 year of experience with … enable end-to-end fine-tuning for foundational GenAI models and optimize them for production environments, including specialized hardware like TPUs. 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. ...

problem-solving skills with the ability to work independently and collaboratively. Clear written and verbal communication skills. Familiarity with deep learning frameworks (TensorFlow, PyTorch, JAX) advantageous. Experience with optimisation, reinforcement learning, or LLMs a plus. Exposure to chip design or applied AI research desirable. Keywords: Research Engineer/Machine … Learning/AI/LLMs/Reinforcement Learning/Generative Models/Deep Learning/PyTorch/TensorFlow/JAX/Python/C++/Semiconductor/Chip Design/Optimisation/London/Cambourne/UK/Permanent If you are interested in this Research Engineer position, please ...

DeepMind • London, UKBack to jobs1. Home2. Jobs3. UK4. Research Scientist, Agent Post-Training5h agoD## Research Scientist, Agent Post-TrainingDeepMind00ResearchFull-timeLondon, UKApply Job### Job DescriptionAt Google DeepMind, we've built a unique culture and work environment where long-term ambitious research can flourish. In this role, you will push … reinforcement learning. You will value various experience and backgrounds to create extraordinary impact.Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high ...

have built and shipped ML systems used by real users. Examples of relevant experience include: Model training and evaluation Production inference systems PyTorch or JAX GPU-based training and inference ML experimentation and iteration Monitoring and debugging production ML systems AI reliability and performance optimisation End-to-end ownership … production Independent ownership and execution Excellent problem-solving ability A bias towards shipping, measuring outcomes, and iterating quickly Technical Environment Python PyTorch/JAX GPU-based training and inference systems OpenAI, Anthropic and open-source models SQL/NoSQL Docker Kubernetes AWS, Azure or GCP Compensation Compensation is intentionally ...

make that hardware serve large AI models at rack scale by building and extending the runtime and serving stack that connects PyTorch and JAX down to the metal. This is a whole-stack systems role. You’ll work where the runtime, the network, and the accelerator meet, partnering closely … cache and memory management, and reliability across distributed failure domains at cluster scale. Fluency at the framework boundary, connecting accelerators to PyTorch/JAX and serving stacks without treating either as opaque. Whole-stack debugging: end-to-end and timeline tracing, workload replay, and reasoning from architectural constraints (SRAM ...

systems Strong Python software engineering skills Experience training, fine-tuning or deploying modern machine learning models Strong experience with PyTorch and/or JAX Experience building scalable ML infrastructure and inference pipelines Comfortable owning projects end-to-end in fast-moving environments Previous mentoring or technical leadership experience would … advantageous Tech Stack Python * PyTorch * JAX * GPU Training * LLMs * Modern ML Infrastructure Get in touch for more details - McGregor Boyall is an equal opportunity employer and do not discriminate on any grounds. ...