16 of 16 Remote PyTorch Jobs in Cambridge

and GenAI technologies for specific scenarios, estimate the time, cost and outcomes of different solutions, and apply effective risk management to ensure prompt delivery Deep experience in Python and PyTorch/TensorFlow. Industry experience in building scalable, production-grade AI services and automated data pipelines with tools like Docker, Kubernetes, and cloud platforms (e.g., AWS). Proven ability to collaborate More ❯

programming skills in languages such as Python with a strong focus on code optimization, modular design, and efficiency. Deep expertise in developing, training, and deploying models using frameworks like PyTorch and TensorFlow. Demonstrated experience in building scalable, production-grade AI services and automated data pipelines with tools like Docker, Kubernetes, and cloud platforms (e.g., AWS). Strong understanding of data More ❯

programming skills in languages such as Python with a strong focus on code optimisation, modular design, and efficiency. Deep expertise in developing, training, and deploying models using frameworks like PyTorch and TensorFlow. Demonstrated experience in building scalable, production-grade AI services and automated data pipelines with tools like Docker, Kubernetes, and cloud platforms (e.g., AWS). Strong understanding of data More ❯

field; or equivalent research experience in industry or academia. Open to applicants across a range of seniority levels with relevant expertise. Core Skills: Proficiency with ML frameworks such as PyTorch, TensorFlow, or JAX. Experience in efficient ML techniques (e.g., quantization, pruning, sparsity, distillation). Practical knowledge of AI deployment on embedded/mobile platforms, including CPUs, GPUs, and NPUs. Hands More ❯

field. Strong familiarity with Linux and the open-source ecosystem. Proficient experience working with machine learning and large datasets. In-depth understanding of open source machine learning frameworks (e.g., PyTorch, ggml, llama.cpp, vllm). Experience building complex systems on the cloud. Experience building and optimizing distributed systems and large-data applications, including those using tensor accelerators or GPUs. Strong analytical More ❯

languages such as Python and shell scripting with a strong focus on code optimization, modular design, and efficiency. Demonstrable experience in developing, training, and deploying models using frameworks like PyTorch and TensorFlow. Strong understanding of data analysis, model evaluation, and error analysis to drive continuous model improvement. Excellent communication skills. Demonstratable commitment to continuous learning, staying current with state-of More ❯

equivalent in a technical discipline (e.g., computer science, chemistry, physics, engineering). Strong background in machine learning, including deep learning and generative models. Experience with ML frameworks such as PyTorch, TensorFlow, or JAX. Proficiency in Python and C++, with a collaborative mindset. Familiarity with protein structure modeling, co-folding, or related methods; knowledge of organic chemistry is a plus. Why More ❯

in CS/EE or related research experience in academia or industry We will consider various levels of experience in relevant research areas Key Skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU, NPU) Experience with distributed More ❯

HW and SW design Developing and working with large software systems in programming languages like Python Knowledge of state-of-the-art deep learning libraries such as Tensorflow and Pytorch Training large deep learning models on powerful GPU-based systems, "Nice To Have" Skills and Experience: ML Model Optimization techniques targeted for resource constrained ARM edge compute platforms. In Return More ❯

others to succeed. "Nice To Have": Experience in Cybersecurity technology projects (Plus!). Experience in the development of ML/AI applications using state-of-the-art toolchains, i.e. Pytorch, TensorFlow (Plus!). Experience with HW and SW Architecture Modelling. Model-Based System Engineering, ISO 15288, MagicGrid Framework, SysML 2.0 (Plus!). Exposure and familiarity with Reliability methodologies. Knowledge of More ❯

benchmarking. Proficiency in Git and distributed version control systems. Desirable Skills Experience with open-source software maintenance, SDN, Kubernetes, and Docker. Hands-on testing of AI networking stacks (e.g., PyTorch, RDMA/RoCE, NCCL). Performance benchmarking and multi-node setup testing using traffic generators like IXIA. Accommodations at Arm At Arm, we want our people to Do Great Things. More ❯

HW-SW co-design Ability to develop and work with large software systems in programming languages like Python Knowledge of cutting-edge deep learning libraries such as Tensorflow, and Pytorch Willing to learn and train large deep learning models on GPU-based systems Qualities that will help your application stand out: Experience with ML model design, optimization, and HW-SW More ❯

solving and debugging skills Practical, organized and analytical approach to work Good oral and written English skills "Nice To Have" Skills and Experience : Experience with ML software frameworks (e.g. PyTorch) Familiarity with ML hardware accelerators (e.g. NPUs, TPUs, GPUs with CUDA support) Knowledge of optimising and profiling software Experience with assembly programming Software development and integration on Linux, Android, or More ❯

Python and Bash A desire to have a positive impact both within our team and the wider Arm ecosystem Strong understanding of machine learning fundamentals and frameworks such as PyTorch and TensorFlow 'Nice to have' abilities and knowledge: Performance evaluation experience and knowledge of potential pitfalls Knowledge of hardware modelling concepts, including configuration and latency definition files Graphics programming experience More ❯

experience working with global, remote teams. Nice to have skills and experience Experience with Windows WDDM or additional Linux kernel driver work. Familiarity with AI frameworks such as TensorFlow, PyTorch, and ONNX. Knowledge of multimedia applications (video, camera, display, GPU). In return This role offers a chance to showcase technical and leadership skills by engaging with product and design More ❯

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 ❯