AI Research Scientist/Manager - Spatial Reasoning

About the role

At P-1 AI, we are building an AI engineer agent named Archie designed to fit seamlessly into engineering teams. Currently at the junior mechanical and electrical engineer level, Archie uses complex engineering tools and has a quantitative intuition over the product design space.

Responsibilities

• Research and develop new approaches for spatial reasoning, geometric understanding, physical reasoning, and structured world modeling in large-scale AI systems.
• Own the full development pipeline from data generation to evaluation to product integration.
• Contribute to both research strategy and technical implementation—this is a hands-on role.
• Investigate topics such as 3D reasoning, geometric representations, multimodal reasoning, world models, simulation, and physical intuition.
• Bring promising ideas into reality through literature review, conference attendance, and other means.

Requirements

• Have conducted research in one or more areas such as spatial reasoning, geometric deep learning, 3D vision, multimodal reasoning, world models, robotics, simulation, or related fields.
• Be fluent in Python and modern ML stack such as PyTorch or JAX, and distributed training frameworks.
• Have a creative approach to bringing definition and solutions to under-specificed challenges.
• Be fascinated by how intelligent systems perceive, represent, and reason about the physical world.
• Thrive in fast-moving, collaborative environments, and communicate technical matters with clarity.
• Take a high-ownership, "make it work" approach to problem solving.

Qualifications

• A PhD (or equivalent experience) in Computer Science, Robotics, Engineering, Math, or a related field.
• Publications in top-tier venues.
• Experience working with CAD data, point clouds, meshes, simulation environments, or other geometric representations.

Skills

• Strong background in spatial reasoning, geometric deep learning, 3D vision, multimodal reasoning, world models, robotics, simulation, or related fields.
• Proficiency in Python and modern ML stack such as PyTorch or JAX.
• Experience with distributed training frameworks.
• Ability to creatively solve under-defined challenges.
• Interest in how intelligent systems perceive, represent, and reason about the physical world.
• High ownership and problem-solving orientation.

Benefits

• Competitive salary
• Meaningful equity ownership
• Healthcare, dental, vision
• 401(k) match
• Unlimited PTO

Pay

Competitive salary

Schedule

Onsite in San Mateo, CA. Relocation support available.