13 of 13 Remote/Hybrid Google JAX Jobs in London

computer science, or a related field. Deep practical experience in Python and at least one modern machine learning framework such as PyTorch, TensorFlow, or JAX, experience working with large compute clusters and ML infrastructure is a plus. A track record of leading self-directed research projects that go well ...

Strong Python development skills Deep Learning (essential) Building and developing Deep Learning models from research through to implementation System Modelling or Simulation Modelling PyTorch, JAX, NumPy or similar machine learning frameworks Experience working within an Applied Research or R&D environment Desirable experience includes: Time Series modelling or forecasting ...

language understanding. The ability to write clean and maintainable code in Python, Rust, C++ or Java Experience with frameworks such as PyTorch, Tensorflow or JAX Strong written and verbal communication skills to document your work effectively and present it to relevant stakeholders and users A passion for identifying ...

Advanced proficiency in Python and C++ within a Linux/UNIX environment. Technical Stack: Extensive practical experience with leading machine learning frameworks (e.g., PyTorch , JAX , or TensorFlow ). Research Experience: A demonstrated track record of designing and implementing complex machine learning architectures to solve rigorous, data-heavy problems. Note ...

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

strong programming skills in Python , Rust, or similar; and strong software engineering fundamentals building performant and reliable systems. Proficient in deep learning frameworks (Pytorch, JAX, TensorFlow). You can work on different layers of the stack from low-level training backends, data ingestion to ML/RL algorithmic design ...

reinforcement learning Experience with agent frameworks, tool-use planning and workflow orchestration Systems-level fluency across data pipelines, training infrastructure and inference (PyTorch, JAX, TensorFlow) Strong MLOps discipline: CI/CD for models, offline/online evaluation, safety and guardrails The ability to influence at exec level and mentor ...

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

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

operating production ML systems Comfortable working across models, infrastructure and product Enjoy working in fast-moving, early-stage environments Tech Stack Python * PyTorch * JAX * LLMs * vLLM * Vector Databases * Modern Agent Frameworks Get in touch for more details - McGregor Boyall is an equal opportunity employer and do not discriminate ...

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

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

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