6 of 6 Google JAX Jobs in London

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

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

outside IR35, for a major pharmaceutical company. The key skills required for this MLOps Engineer position are: MLOps Image data Terraform 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. ...

outside IR35, for a major pharmaceutical company. The key skills required for this MLOps Engineer position are: MLOps Image data Terraform 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 click apply ...