51 to 75 of 104 Google JAX Jobs in London

running experiments, and debugging pipelines, and who can carry research results through to production with engineering. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow). Ability to communicate clearly, collaborate across teams, and align research work with product and engineering priorities. Nice to have Experience quantifying ...

experience with other robotics simulators (Drake, MuJoCo, Newton, Isaac, RaiSim, Jiminy, Brax). PyTorch, our user-facing interface. Other JIT compilers such as Triton, JAX, or Numba. 3D tooling and interactive editors (Maya, Blender, Houdini). Serialization formats (USD, glTF, MJCF/URDF). C++, CUDA, or Metal. Developer ...

research direction with strong execution, and experience mentoring others on a fast‐moving, applied research team. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Nice to have Demonstrated experience fine‐tuning ...

skills. Ability to balance strategic thinking with hands-on technical leadership. Comfortable operating within a fast-paced, high-growth environment. Technology Environment Python PyTorch JAX Large Language Models (LLMs) GPU Training & Inference Infrastructure Distributed ML Systems Why Join? Help shape the future of AI-powered productivity. Work alongside ...

models — architecture, training, and inference. Experience solving real problems using reinforcement learning with deep neural networks in any domain. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You are self‐driven, pro‐active, communicate efficiently, document experiments clearly and communicate trade ...

bitwidth numerics. You will also be expected to assist in the design of novel hardware architectures for accelerating our models, mapping AI models from JAX graphs to transistors. Attributes You will be an experienced software engineer, likely in C/C++/Rust or a similar language. You will ...

easy or obvious solutions.QualificationsStrong engineering skills, especially any of: CUDA/Triton/Pallas/CuTe DSL kernel development, lower-level PyTorch/JAX/XLA development, CUDA Graphs, FPGA/ASIC experienceMust have two or more years work experience building deep learning systems, for any domain: robotics, biology ...

lead complex research efforts while staying grounded in product impact, user experience, and real-world performance. Strong coding and experimentation skills (Python, PyTorch/JAX, audio processing libraries). Ability to communicate clearly, collaborate across teams, and align research work with product and engineering priorities. Proven experience mentoring others ...

integrating ML systems into production while staying grounded in product impact and real-world quality. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Ability to lead complex research efforts, to communicate clearly ...

C++ or Java A strong background in the mathematical foundations of core ML algorithms, paired with hands‐on experience using deep learning frameworks (PyTorch, JAX, TensorFlow), agentic orchestration ecosystems (LangGraph, Google ADK), and Model Context Protocol tooling (RustMCP) Strong written and verbal communication skills to document your work ...

evaluation, and testing Optimise reliability, scalability, latency, and user experience Work closely with Product and Engineering teams to solve complex customer problems Python PyTorch JAX vLLM Vector Databases What We're Looking For Strong software engineering fundamentals Experience building AI agents, copilots, RAG systems, or workflow automation platforms Excellent ...

integrating ML systems into production while staying grounded in product impact and real‐world quality. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Ability to lead complex research efforts, to communicate clearly ...

models, fighting deep learning compilers). Excellent understanding of GPU architecture and why some models run faster than others. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice to have: Robotics ...

impacting real world drug design programs.Skills and qualifications Essential:Significant experience with large scale distributed training of LLMs.Experience with deep learning ML frameworks (either JAX or PyTorch).Knowledge of parallelism strategies and collective communication libraries (e.g. NCCL).Good understanding of GPU architectures. Reasoning about performance concepts is more important ...

design and commitment to the development of reusable, scalable, and testable software components. Basic understanding of at least one major deep learning framework (PyTorch, JAX, TensorFlow). Knowledge of command‐line tools and shell scripting. Knowledge of software engineering best practices, including continuous integration (CI) and continuous deployment … issues in live systems. Experience designing and administering SQL databases, schema design, query performance, and day‐to‐day operational management. Hands‐on experience with Google Cloud Platform, in particular the services we build on: Cloud Run, Google Kubernetes Engine, Cloud Storage, Artifact Registry, Cloud SQL. Fluency in English. ...

with strong software engineering practicesDemonstrated expertise in building production-quality code and data pipelines for ML systemsProficiency in modern AI development frameworks including: PyTorch, Jax , HuggingFace Transformers, LLM APIs (litellm etc) and vLLM for building and deploying large-scale AI applicationsUnderstanding of LLM training methodologies including instruction fine-tuning … data generation, data curation, or curriculum learning pipelinesExperience training large-scale models over distributed nodes with cloud tools such as AWS, MS Azure, or Google CloudHands-on experience with MLOps, experiment tracking, and model deployment systemsStrong interest in staying current with ML research literature and ability to quickly implement ...

data quality, and ensure continuous training cycles. Preferred qualifications include 3+ years in deep learning, proficiency in Python and frameworks like PyTorch/JAX, and effective communication. The role offers 23 days off, private healthcare, and a pension scheme. #J-18808-Ljbffr ...

from data to UI. Applicants should have strong quantitative skills and experience in Python/SQL/TypeScript and PyData stack, plus PyTorch/JAX/TF for model development. #J-18808-Ljbffr ...

performance, with no prior finance requirement. Candidates hold a Master’s or PhD in ML, CS, or statistics, and excel in Python, Torch/JAX/TF, #J-18808-Ljbffr ...

live training/matchday schedule, not just be technically neat. Python/SQL/TypeScript, PyData stack (Pandas/Polars, Altair, etc.), PyTorch/JAX/TF. You'll own things end-to-end, data through UI. Looking for strong quant background, first-principles engineering judgment, and taste ...

have strong experience building production-grade machine learning systems at scale You are experienced with modern ML frameworks such as PyTorch, TensorFlow, or JAX You have worked with or are interested in multimodal machine learning You understand how to design systems that balance automation with quality, safety, and user ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable ...

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, ...)Excellent communication skills to report and present research findings and developments clearly, both orally and in writing.Curious and innovative disposition capable … 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 ...

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, ...). Excellent communication skills to report and present research findings and developments clearly, both orally and in writing. Curious and innovative disposition … writing production code. Experience training large-scale models over distributed nodes with cloud tools and providers such as Amazon AWS, Microsoft 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 ...

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 … 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 ...