Software Engineer, ML Platform

Job summary

Responsibilities

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.
• 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.
• Design, build, and operate core platform systems used daily by ML researchers and product engineers

Tech stack

• Kubernetes
• Ray
• Spark