Graduate Research Scientist, AI

Arm is a company focused on shaping the future of energy-efficient AI through innovative research. As a Graduate Research Scientist, you will be part of a global team working on foundational advances in machine learning, computer architecture, and systems, with the goal of influencing billions of devices worldwide.

Responsibilities

• Scalable and Distributed Machine Learning: Approaches for training, inference, and deployment of increasingly capable models across diverse and distributed compute environments
• Efficient and Sustainable AI: Techniques to improve the performance, efficiency, and scalability of machine learning workloads, spanning algorithms, representations, and system-level optimizations
• Generative Modeling: Research into modern generative methods for learning, representation, and synthesis across different data types and tasks
• Multimodal and Integrated Learning: Models and systems that combine multiple sources of information to enable richer perception, understanding, and interaction
• Autonomous and Adaptive Systems: Learning-based methods for reasoning, decision-making, coordination, and adaptation in complex or dynamic environments
• Embodied and Interactive Intelligence: Machine learning approaches for perception, planning, and control in systems that interact with the physical world or with users

Skills

• Hold (or are on track to hold) a PhD in Computer Science, Electrical/Computer Engineering or related field by spring 2026
• Candidates with alternative degrees will also be considered if they have relevant experience
• Proficiency in programming languages such as Python, C/C++, or similar
• Experience designing, training, or analyzing machine learning models or systems using standard frameworks (e.g. PyTorch, Hugging Face Transforms, etc.)
• Interest in research and long-term innovation beyond immediate product delivery
• Contributions to open-source and/or peer-reviewed publications
• Up to date understanding of machine learning trends
• Familiarity with machine learning model optimization or deployment
• Familiarity with Arm architecture
• Familiarity with GPU programming

Benefits

• Competitive salary and comprehensive rewards package
• Support, autonomy and opportunity to excel in your career
• Mid-year, and year-end review
• Grad-teach-Grads workshops
• Guest speaker series
• Various social events
• Exclusive Global Graduate Conference (GGC)

Company Overview

Company H1B Sponsorship