TPUs Jobs

Datadog is seeking a customer-focused, deeply technical Distinguished Architect to join our Product Solutions Architecture team. In this role, you will act as a technical multiplier for leading AI labs and AI-native companies, bridging the gap between their infrastructure aspirations and Datadog’s technology roadmap. You will ensure our platform natively solves the unique observability challenges of training and deploying foundational models at scale. We operate as a hybrid workplace to foster collaboration and creativity while ensuring work-life harmony.

$330k - $440k

Lightmatter is seeking a Principal Machine Learning Infrastructure Researcher to join our Machine Learning team. This team is focused on inventing novel hardware systems using Lightmatter technology and modeling the performance of Lightmatter-enabled systems on both training and inference workloads. This is primarily a creative role where you need to think outside the box to push the boundaries of systems research with the power of silicon photonics at your fingertips. This role will work closely with our product, solutions architecture, and engineering teams to generate impactful performance analysis, create high-performance rack-scale designs, and publish cutting-edge ML systems research.

$300k - $350k