About Usamah Zaheer
I'm Usamah Zaheer, a Machine Learning Software Engineer at Arm in Cambridge, UK, where I design and optimise ML infrastructure for Arm architectures. My work focuses on making deep learning models run faster and more efficiently — from optimising ML compilers and libraries to conducting deep kernel-level analysis that eliminates inference bottlenecks.
Before Arm, I spent two years at Dyson as a Robotics Software Engineer. There, I developed CNN algorithms for segmentation, object detection, and classification — deploying them on robot hardware using techniques like quantisation, pruning, and knowledge distillation. I also built a VLM solution that saved over £100,000 and boosted productivity by 20x. One of the highlights was presenting my work directly to the CEO and senior leadership.
My journey in ML research started at the University of Leicester, where I worked as an ML Research Assistant. I built end-to-end automated ML pipelines for processing high-resolution satellite imagery in real-time, integrating CNNs in PyTorch and TensorFlow. I also led development of predictive models using Random Forest and SVM, and deployed AI solutions in cloud environments with Docker and Kubernetes.
I'm currently pursuing an MS in Artificial Intelligence at the University of Texas at Austin, building on my MS in Embedded Systems and Control Engineering from the University of Leicester (where I graduated with Distinction) and my BTech in Electronics and Communication Engineering from Jawaharlal Nehru Technological University.
My technical toolkit spans the full ML stack: PyTorch, TensorFlow, JAX, CUDA, TensorRT, ArmNN, and C++ for performance-critical work. I'm passionate about the intersection of ML and hardware — making models not just accurate, but fast and deployable at the edge.
Between Arm and starting my MS at UT Austin, I led the design and implementation of AI agents at a stealth startup, integrating LLMs and NLP for scalable solutions using RAG, LangChain, and LlamaIndex. I managed cloud infrastructure on Vertex AI and oversaw containerised deployments with Docker and Kubernetes.
Outside of work, I mentor undergraduates in robotics and machine learning, participate in hackathons, and contribute to open-source projects. You can find my work on GitHub, connect with me on LinkedIn, or follow me on X (Twitter).
Explore my projects to see what I've built, or check out my blog where I write about machine learning, edge computing, and software engineering.
Frequently Asked Questions
Who is Usamah Zaheer?
Usamah Zaheer is a Machine Learning Software Engineer at Arm, based in Cambridge, UK. He has over 4 years of experience in deep learning, computer vision, edge computing, and MLOps. He previously worked at Dyson as a Robotics Software Engineer.
What does Usamah Zaheer work on?
Usamah works on designing and optimising ML infrastructure for Arm architectures. He specialises in ML compilers, inference performance, deep learning frameworks (PyTorch, TensorFlow, JAX), and deploying models on edge devices.
Where did Usamah Zaheer study?
Usamah is currently pursuing an MS in Artificial Intelligence at the University of Texas at Austin. He holds an MS in Embedded Systems and Control Engineering from the University of Leicester (Distinction) and a BTech in Electronics and Communication Engineering from Jawaharlal Nehru Technological University.
Where has Usamah Zaheer worked?
Usamah has worked at Arm (Machine Learning Software Engineer), Dyson (Robotics Software Engineer), the University of Leicester (ML Research Assistant), and a stealth startup (AI Agent Systems). He has also contributed to open-source projects.