Research Scientist (Visual Generative AI & World Models) at Graphcore

Graphcore is seeking a researcher to explore the intersection of visual generative models, world models, and hardware-aware machine learning. You will work within the SoftBank Group ecosystem to influence the future of AI compute by bridging the gap between frontier research and silicon architecture.

Key facts

- Location: Cambridge, UK
- Engagement: Full-time
- Team: Graphcore Research

What you'll do

- Design and execute research projects in visual generative AI, multimodal learning, and world models.
- Manage the full research lifecycle, from hypothesis and implementation to analysis and publication.
- Present findings at major machine learning conferences and workshops.
- Collaborate with software and silicon engineering teams to align emerging AI workloads with future hardware systems.

Requirements

- Master degree, PhD, or equivalent professional experience in a technical field such as Mathematics, Statistics, Computer Science, Physics, Chemistry, or Biomedical Engineering.
- Proven background in visual generative AI, world models, or visual understanding.
- Proficiency in Python and contemporary deep learning frameworks like JAX or PyTorch.
- Solid understanding of deep learning theory,



including model scaling, optimization, and architecture design.
- Demonstrated research output through publications, technical reports, open-source contributions, or industrial research projects.
- Mathematical proficiency in linear algebra, probability theory, and calculus.

Nice to have

- Experience with embodied AI, robotics, action-conditioned models, or multimodal generation.
- Low-level programming skills in CUDA, C++, or Triton.
- Understanding of hardware-level deep learning constraints, including memory hierarchy, bandwidth, parallelism, and matrix engines.
- Knowledge of deep learning software stacks, including kernel fusion, graph compilation, XLA/ATen operations, and asynchronous execution.

Skills & tools

- Python - PyTorch - JAX - Deep Learning Frameworks - Research Methodology

Applicants must possess the legal right to work in the UK. Graphcore cannot provide visa sponsorship for this position.

Benefits include a competitive salary, pension matching up to 5 percent, private medical insurance, dental plan, life assurance, income protection, and flexible working arrangements.

#J-18808-Ljbffr

📌 Research Scientist (Visual Generative AI & World Models) (Cambridge)
🏢 Graphcore
📍 Cambridge