and hardware, developing innovative solutions, and collaborating with teams to build high-performance AI systems. Responsibilities: Your responsibilities involve working with major ML frameworks (PyTorch, TensorFlow, etc.) to port and develop ML networks, optimize and quantize models for efficient execution on Arm platforms, and help ensure multiple Arm products are … Required Skills and Experience: • A background in computer science, software engineering or other comparable skills • Experience training and debugging neural networks with TensorFlow and PyTorch using Python • Understanding, deploying, and optimizing Large Language Models (LLMs) and Generative AI algorithms. • Experience using software development platforms and continuous integration systems • Familiarity with … investigate are well understood and the critical areas needing improvement are understood Nice To Have Skills and Experience: • Experience of the inner workings of Pytorch, Tensorflow, Executorch and Tensorflow Lite • Experience of developing and maintaining CI/testing components to improve automation of model analysis • Good knowledge of Python for More ❯
independently, define goals and scope and lead your own development effort. KEY RESPONSIBILITIES: Optimize Deep Learning Frameworks: Enhance and optimize frameworks like TensorFlow and PyTorch for AMD GPUs in open-source repositories. Develop GPU Kernels: Create and optimize GPU kernels to maximize performance for specific AI operations. Develop & Optimize Models … Triton for multi-GPU and multi-platform performance. Deep Learning Integration: Proficient experienced in integrating optimized GPU performance into machine learning frameworks (e.g., TensorFlow, PyTorch) to accelerate model training and inference, with a focus on scaling and throughput. Software Engineering: Proficient skills in Python and C++, with experience in debugging More ❯
independently, define goals and scope and lead your own development effort. KEY RESPONSIBILITIES: Optimize Deep Learning Frameworks: Enhance and optimize frameworks like TensorFlow and PyTorch for AMD GPUs in open-source repositories. Develop GPU Kernels: Create and optimize GPU kernels to maximize performance for specific AI operations. Develop & Optimize Models … Triton for multi-GPU and multi-platform performance. Deep Learning Integration: Proficient experienced in integrating optimized GPU performance into machine learning frameworks (e.g., TensorFlow, PyTorch) to accelerate model training and inference, with a focus on scaling and throughput. Software Engineering: Proficient skills in Python and C++, with experience in debugging More ❯
days a week in the Cambridge office. This role focuses on maintaining and optimising the Linux operating system, file systems, and software stack (Cuda, PyTorch, Python etc) for machine learning projects as well as setting up and configuring NVIDIA HGX servers (installing and updating software, managing user access, and ensuring … Cloud environments, Experience with NVIDIA GPU technologies, such as NVLink, NVSwitch, and Tensor Core GPUs, Experience with machine learning frameworks and libraries, such as PyTorch and associated system optimisations, Experience with NAS servers, Experience with data version control systems. Benefits we offer: 23 days' holiday + all public holidays, rising More ❯
days a week in the Cambridge office. This role focuses on maintaining and optimising the Linux operating system, file systems, and software stack (Cuda, PyTorch, Python etc) for machine learning projects as well as setting up and configuring NVIDIA HGX servers (installing and updating software, managing user access, and ensuring … Cloud environments, Experience with NVIDIA GPU technologies, such as NVLink, NVSwitch, and Tensor Core GPUs, Experience with machine learning frameworks and libraries, such as PyTorch and associated system optimisations, Experience with NAS servers, Experience with data version control systems. Benefits we offer: 23 days' holiday + all public holidays, rising More ❯
