AI Theory

About the role

As a Member of Technical Staff, you will be a foundational member of our small, multi-disciplinary R&D team. We are looking for 'first principles' thinkers who are excited to tackle the hardest, most ambiguous technical challenges at the intersection of AI, physics, and computer architecture.

Responsibilities

• Drive invention, prototyping, and validation of the core components of our novel computing platform.
• Work on a wide range of tasks, spanning from theoretical modeling and simulation to algorithm development, hardware/software co-design, or experimental validation in collaboration with other team members.
• Navigate deep uncertainty and help chart our technical roadmap.

Requirements

• Exceptional technical ability in a quantitative field (e.g., Physics, Computer Science, Electrical Engineering, Applied Math, or a related discipline).
• An MS/PhD or equivalent research/project experience is strongly preferred.
• A "0-to-1" mindset. You have a demonstrated history of tackling complex, ambiguous R&D problems, often from a blank slate.
• Deep curiosity. You are comfortable diving into new domains, whether it's semiconductor physics, machine learning theory, or systems-level design.
• A creative and unconventional approach to problem-solving.

Qualifications

• Analytic Foundations: Core competences in the analysis of nonlinear dynamical systems (ODEs, PDEs, SDEs), ideally with experience analyzing the stability, noise robustness, and capacity of such systems.
• Practical Eye: The ability to leverage analytic insights to build practical tools – metrics, algorithmic optimizations, and automated analyses – that can be used to study dynamical systems.
• Programming Proficiency: Strong command of Python and expertise with using numeric computing and visualization libraries, such as numpy, scipy, and matplotlib. Experience with libraries geared towards analysis, such as computer algebra libraries (e.g., sympy) also recommended.
• ML/AI Familiarity: Familiarity with dynamics-based ML model architectures, such as diffusion models and energy-based models, and general experience with ML model training flows. Experience with using high-level ML model frameworks, such as PyTorch and JAX.

Skills

• Exceptional technical ability in a quantitative field (e.g., Physics, Computer Science, Electrical Engineering, Applied Math, or a related discipline).
• An MS/PhD or equivalent research/project experience is strongly preferred.
• A "0-to-1" mindset. You have a demonstrated history of tackling complex, ambiguous R&D problems, often from a blank slate.
• Deep curiosity. You are comfortable diving into new domains, whether it's semiconductor physics, machine learning theory, or systems-level design.
• A creative and unconventional approach to problem-solving.

Benefits

• A comprehensive package including best-in-class health benefits, 401k matching, truly unlimited PTO, and complimentary meals in our Palo Alto office.

Pay

Competitive salary commensurate with experience.

Schedule

Full-time, remote work option available.