2506 IT & Software Developer jobs in the UK
Senior PyTorch Engineer
£32,000 - 72,000
Advanced Micro Devices
Cambridge Science Park 196, Eastern England
£32,000 - 72,000
Requirements
Must:
- Masters or PhD level degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field
- Expert skills in Python and C++
- Experience in debugging, performance tuning, and test design
- Strong understanding of AI frameworks such as PyTorch, Triton, and vLLM
- Applied knowledge in Natural Language Processing, Vision, Audio, and Recommendation Systems
- Strong experience with GPU Programming models (CUDA, HIP)
- Expert with modern software practices and tooling (e.g., debuggers, profilers, GitHub)
- Effective communicator with strong analytical and problem-solving skills
- Senior professional experience in technical software development, focusing on GPU optimization and performance engineering
Responsibilities
- Lead the development and optimization of Deep Learning models on AMD GPUs
- Enhance PyTorch and its ecosystems on AMD GPUs with performance improvements
- Collaborate with PyTorch maintainers to align on requirements and propose enhancements
- Work with community partners to enable and upstream new features related to AMD GPUs
- Identify performance bottlenecks by partnering with internal GPU Libraries and runtime teams
- Develop innovative solutions for multi-GPU and multi-node distributed training environments
- Apply rigorous software engineering best practices including code review, testing, and CI/CD integration
Description
At AMD, we are on a mission to build great products that accelerate next-generation computing experiences, from AI and data centers to PCs and gaming. We pride ourselves on our culture of innovation and collaboration, driving progress through bold ideas and human ingenuity. Our team is composed of incredibly talented industry specialists, working with the latest hardware and software technology. We offer an engaging work environment where you can shape the future of AI and advance your career. Join us and become part of our diverse and inclusive team.
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 2506 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?
City of Lincoln Council, Bluefruit Software Limited, Openda Ltd, National Highways, Learning Partners, Counter Terrorism Policing, OKTO Integrated Services among others, are currently hiring for ML, AI roles in the UK.
The company with most openings is Experis as they are hiring for 75 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 Experis as they are hiring for 75 different Machine Learning Engineer jobs in the UK. They are probably quite committed to find good Machine Learning Engineers.