[ engineering ]

Staff Software Engineer - Ai Research Infrastructure

Databricks
Location
New York City, New York; San Francisco, California
Work mode
Remote
Type
full time
Listed

Applications go directly to Databricks. Ramai takes no fee for placement and no money for ranking.

[ About this role ]

Staff Software Engineer on the AI Research Infrastructure team at Databricks. This role involves designing and building the research stack that powers Databricks AI Research, including services for scheduling, orchestrating, and observing large-scale training and inference workloads across thousands of GPUs. The engineer will also improve developer tooling to enhance researcher productivity and reliability.

Skills and focus areas

  • distributed systems
  • gpu computing
  • experiment management
  • ci/testing
  • large-scale systems
  • reliability
  • efficiency
  • security
  • software design
  • software implementation
  • research infrastructure

Summary written by Ramai from the original posting. Full details and requirements are on Databricks’s careers page.

[ More at Databricks ]

See all 901 open roles at Databricks →

[ Similar roles at other startups ]

Browse all open engineering roles →

[ Not quite right? ]

Tell the assistant what you’re after — stage, location, stack, visa sponsorship — and it searches every open role for you. A few questions are free, no account needed.