192 IT & Software Developer jobs in the UK
Senior PyTorch Engineer - Eastern England
£32,000 - 72,000
Advanced Micro Devices
Cambridge Science Park 196, Eastern England
£32,000 - 72,000
Requirements
Must:
- Expert skills in Python and C++
- Strong understanding of AI frameworks such as PyTorch, Triton, and vLLM
- Applied knowledge across domains like Natural Language Processing, Vision, Audio, and Recommendation Systems
- Strong experience with GPU programming models (CUDA, HIP)
- Expert with modern software practices and tooling, including debuggers, profilers, and version control systems (e.g., GitHub)
- Effective communicator with strong analytical and problem-solving skills
- Masters or PhD in Computer Science, Computer Engineering, Electrical Engineering, or a related field
- Senior professional experience in technical software development focusing on GPU optimization, performance engineering, and framework development
Responsibilities
- Lead the development and optimization of Deep Learning models on AMD GPUs
- Enhance PyTorch and its ecosystems on AMD GPUs, contributing performance-critical improvements to upstream repositories
- Engage with PyTorch and other open-source maintainers to align on requirements and propose enhancements
- Collaborate with the PyTorch community and ecosystem partners to design and enable new features
- Partner closely with internal GPU Libraries, compilers, and runtime teams to identify performance bottlenecks
- Develop innovative solutions for scale-up (multi-GPU) and scale-out (multi-node) distributed training environments
- Apply rigorous software engineering best practices, including code review, testing, CI/CD integration, and debugging
Description
At AMD, we are committed to building great products that revolutionize computing experiences. Our team thrives on innovation and collaboration, believing that progress stems from bold ideas and diverse perspectives. We offer a vibrant culture that supports personal and professional growth, along with flexible working arrangements. Join us in our mission to shape the future of AI and beyond.
Something wrong or incorrect with this job? Tell us in the chat 💬 on the right ➡️
You can find Machine Learning Engineer salaries in the UK here.
How many Machine Learning Engineer jobs are in the UK?
Currently, there are 192 ML, AI openings. Check also: TensorFlow jobs, Python jobs, Computer-Vision jobs - all with salary brackets.
Is the UK a good place for Machine Learning Engineers?
The UK is one of the best countries to work as a Machine Learning Engineer. It has a vibrant startup community, growing tech hubs and, most important: lots of interesting jobs for people who work in tech.
Which companies are hiring for Machine Learning Engineer jobs in the UK?
SwapTix, Arctic Shores, TBG Solutions Ltd, Companies House, FCDO Services, MOD - Defence Digital, Counter Terrorism Policing among others, are currently hiring for ML, AI roles in the UK.
The company with most openings is Adecco as they are hiring for 6 different Machine Learning Engineer jobs in the UK. They are probably quite committed to find good Machine Learning Engineers.
The company with most openings is Adecco as they are hiring for 6 different Machine Learning Engineer jobs in the UK. They are probably quite committed to find good Machine Learning Engineers.