Want to know if this job is worth applying to?
San Francisco, California, United States
Onsite
Our mission is to automate coding. The first step in our journey is to build the best tool for professional programmers, using a combination of inventive research, design, and engineering. Our organization is very flat, and our team is small and talent dense. We particularly like people who are truth-seeking, passionate, and creative. We enjoy spirited debate, crazy ideas, and shipping code.
As a Software Engineer on ML Platform at SpaceXAI, you'll build the infrastructure that turns real product usage into better models — and keeps research moving fast on large GPU fleets. ML Platform is organized into four teams. Depending on your background, you may join any of them:
Telemetry — Own the collection and serving path that turns real product use into a record research can trust; without slowing the product, and under a small, explicit policy. Client-side or high-volume ingestion experience is a plus.
ML Data Platform — Build the shared environments and pipeline substrate researchers extend, so new experiments don’t fork their own stack.
Observability — Make it easy for researchers to start, watch, and debug their own runs.
ML DevX and Systems — Shorten the path from idea to a trusted run on the research fleet.
We're looking for strong distributed-systems and infrastructure engineers who want to sit next to research and ship platform primitives that move the product.
We're in-person with cozy offices in North Beach, San Francisco, Palo Alto, and Manhattan, New York, complete with well-stocked libraries.
Design, build, and operate core platform systems used daily by ML researchers and product engineers
Partner closely with research to turn recurring pain into durable infrastructure
Own reliability, performance, and developer experience for the systems in your lane
Ship iteratively in a flat, high-ownership environment. Measure impact, then raise the bar
You have a strong background in systems / infrastructure software engineering and enjoy building platforms other engineers depend on
You've owned production distributed systems at meaningful scale (ingestion, data pipelines, scheduling/orchestration, or similar)
You're comfortable across Linux, cloud and/or bare metal, and modern orchestration (Kubernetes, Ray, or equivalent)
You like working closely with ML researchers and product engineers
You thrive where ownership is high and the feedback loop is short
Telemetry: event ingestion, product analytics pipelines, OpenTelemetry / tracing, reliable data APIs
Product Data Platform: data frameworks, Spark / Flink / Ray, ML dataset and training-data infrastructure
Observability: experiment / run monitoring, debug and eval tooling, agent-friendly observability UX
ML DevX and Systems: GPU / cluster scheduling, job queues, node health, research compute developer experience
If there appears to be a fit, we'll reach out to schedule 2-3 short technicals. After, we'll schedule an onsite in our office, where you'll work on a small project, discuss ideas, and meet the team.
