Graduate Research Scientist, AI
Arm is a company that focuses on energy-efficient AI research and the development of intelligent systems. As a Graduate Research Scientist, you will contribute to foundational advances at the intersection of machine learning, computer architecture, and systems, working on scalable and distributed machine learning, efficient AI techniques, and generative modeling.
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
- 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