Cambridge, Cambridgeshire, United Kingdom Hybrid / WFH Options
Arm Limited
for collaborating with remote teams across different time zones. "Nice To Have" Skills and Experience : Experience of Windows WDDM or Linux kernel driver development. Experience of AI frameworks (TensorFlow, PyTorch, ONNX). Knowledge of multimedia use cases including video, camera, display, and GPU. In Return: You will have the opportunity to demonstrate technical expertise and leadership skill to engage with More ❯
vision, bayesian inference, causal reasoning & inference, transfer & multi-task learning, diffusion models, graph neural networks, active learning; Experience writing production-quality code with modern machine learning frameworks such as PyTorch, TensorFlow, JAX, or similar; Experience with multi-GPU and distributed training at scale; A team player who thrives in collaborative environments and is committed to enabling colleagues to reach their … mathematics, statistics, physics, or computational biology, or equivalent experience; Experience in developing machine learning models; Very strong programming skills, including experience with Python and deep learning libraries such as PyTorch or JAX; Experience in large-scale distributed optimization of machine learning models across multiple GPUs and nodes. Preferred Qualifications - Familiarity with biological data formats, concepts, and computational models; Experience in More ❯
solving skills, a proactive approach, and a keen understanding of software engineering best practices are essential. 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: Design and optimize deep learning … Kernel (CK), CUTLASS, and Triton for multi-GPU and multi-platform performance. Deep Learning Integration: E xperience d 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: S kill ed in Python and C++, with experience in debugging, performance tuning, and test More ❯
solving skills, a proactive approach, and a keen understanding of software engineering best practices are essential. 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: Design and optimize deep learning … Kernel (CK), CUTLASS, and Triton for multi-GPU and multi-platform performance. Deep Learning Integration: E xperience d 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: S kill ed in Python and C++, with experience in debugging, performance tuning, and test More ❯
Cambridge, Cambridgeshire, England, United Kingdom
MicroTECH Global Ltd
JOB AD: OverviewWork within our clients machine learning team to deploy and optimize models for applications like low-latency speech recognition and large language models (LLMs). Initial focus will be on improving our clients speech recognition model's training More ❯