9 of 9 JAX Jobs in London

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

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

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

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