12 of 12 JAX Jobs in London

Developer will work with a variety of technologies including the Python scientific stack (NumPy, Pandas, SciPy, etc.), frameworks such as OpenCV, PyTorch, TensorFlow and JAX, data pipeline tools such as Airflow or Prefect and data processing tools such as Spark or Databricks. The Company The Python Developer will be joining ...

machine learning theory, including optimization, generalisation, and various model architectures. Programming : Strong python skills and experience with deep learning libraries (TensorFlow/PyTorch/JAX). Communication : Ability to clearly explain complex ML concepts and research findings to both technical and non-technical audiences. Education : Master's Degree (PhD preferred ...

with a demonstrable track record of models made it from the lab into production environments . Deep Technical Mastery: Expert-level proficiency in PyTorch, JAX, or TensorFlow , with a focus on building custom layers, loss functions, and optimization loops. Published Excellence: A strong record of high-quality publications ...

Mathematics, Physics, Computer Science, or a related field Strong hands-on experience building LLM-based systems with modern ML tooling (PyTorch, Ray, JAX) Experience designing agentic systems (planning, memory, tool use, multi-step reasoning) Strong understanding of transformers and foundation model behavior Comfortable owning whatever needs doing, from research ideas ...

machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down ...

Mathematics, Physics, Computer Science, or a related field Strong hands-on experience building, training, and scaling foundation models with modern ML tooling (PyTorch, Ray, JAX) Strong background in graph neural networks and transformer architectures Demonstrated ability to turn research ideas into robust, scalable implementations Comfortable owning whatever needs doing, from ...

production systems Hands-on post-training experience (RLHF, DPO, or similar) and deployed LLMs in production Strong Python and modern ML tooling (PyTorch/JAX, training infrastructure, evaluation pipelines) Track record shipping research from prototype to production impact Comfortable owning work end-to-end: architecture to data pipelines to production ...

generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. 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 or autonomous ...

skills required for this ML Engineer position are: Building ML systems Computer Vision Building processing pipelines for image data AWS PyTorch, TensorFlow, JAX If you do have the relevant experience for this long term remote contract ML Engineer role, outside IR35, please do apply. ...

skills required for this ML Engineer position are: Building ML systems Computer Vision Building processing pipelines for image data AWS PyTorch, TensorFlow, JAX If you do have the relevant experience for this long term remote contract ML Engineer role, outside IR35, ple click apply for full job details ...

skills required for this ML Engineer position are: Building ML systems Computer Vision Building processing pipelines for image data AWS PyTorch, TensorFlow, JAX If you do have the relevant experience for this long term remote contract ML Engineer role, outside IR35, please do apply. ...

basis, outside IR35, for a major pharmaceutical company. The key skills required for this MLOps Engineer position are: MLOps Image data AWS PyTorch, TensorFlow, JAX If you do have the relevant experience for this long term remote contract MLOps Engineer role, outside IR35, please do apply. ...