Staff Software Engineer, Ray Data

Highlights

• Build core systems powering Ray Data for multimodal AI workloads at scale.
• Optimize distributed execution across pipeline stages in heterogeneous environments.
• Solve scaling challenges with AI-native companies building next-gen AI apps.

What they offer

• Competitive salary and equity
• Health/dental/vision coverage (many plans up to 99% employer-covered)
• Flexible time off
• Paid parental leave
• Mental health support

Ideal candidate

Seasoned distributed systems builder who loves tackling unsolved performance and scale problems and owning core architecture decisions that power next-generation AI workloads.

Requirements

• 6+ years building production-grade software, infrastructure, or developer-facing systems with strong Python
• 6+ years owning core architectural decisions within a distributed data or compute engine
• Deep experience with distributed systems internals such as scheduling, fault tolerance and partitioning
• Track record reasoning through system-level tradeoffs and defending architectural decisions
• Passion for solving unsolved problems in large-scale AI infrastructure