2 of 2 TensorFlow Jobs in Milton Keynes

Generative AI Engineer

Hiring Organisation: SearchWorks
Location: Milton Keynes, Buckinghamshire, UK
Employment Type: Full-time

4+ years of software engineering experience, including experience working with ML/AI systems Strong Python skills with experience using ML frameworks (e.g. PyTorch, TensorFlow, Transformers) Experience building and deploying LLM-powered applications Experience with React, TypeScript, APIs, and cloud-based microservices Hands-on experience with prompt engineering ...

Founding Lead Machine Learning Engineer

Hiring Organisation: A1
Location: Milton Keynes, Buckinghamshire, UK
Employment Type: Full-time

deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills — writing robust, production-grade systems Experience with GPU optimization: memory efficiency ...