Staff Software Engineer - AI Research Infrastructure
Databricks Mosaic · New York City, New York; San Francisco, California · San Francisco, California
CHECKED 15H AGO
Infrastructureonsitestaff$199k–$270k
At a glance
Highlights
- Run research stack orchestrating training and inference across thousands of GPUs.
- Turn experimental AI workloads into robust pipelines shipping models to customers.
- Enable researchers to go from idea to large-scale experiment in minutes, not days.
What they offer
- annual performance bonus
- equity
Ideal candidate
Seasoned infrastructure engineer who loves enabling researchers, thriving at the intersection of systems and ML. Pragmatic, fast-moving and operationally excellent, you mentor others and turn prototypes into reliable large-scale platforms.
Requirements
- BS/MS or PhD in Computer Science or related field
- 5+ years of software engineering experience with large-scale distributed systems or infrastructure
- Deep experience building and operating distributed systems, data pipelines or backend services
- Proficient in systems programming languages and designing complex services
- Built or contributed to cluster schedulers, resource managers or job orchestration systems
- Understand modern ML training and inference workflows
Lab profile
About Databricks Mosaic
Builds unified data analytics and AI platform to productionize AI applications.
What they build
- Databricks platform
- Unified Data Analytics and AI
- AI platform
Known for
- Forward-deployed AI engineering
- Productionizing first-of-kind AI applications
- Customer platform adoption and scale
- Cross-functional engineering and product collaboration