Staff Software Engineer, Ray Data

Responsibilities

• Design, build, and improve the core systems powering Ray Data, focusing on performance, scalability, and reliability.
• Design and optimize distributed execution across data-pipeline stages in heterogeneous environments.
• Build data-loading and processing solutions for production training and inference workloads.
• Solve problems involving distributed execution, scheduling, resource management, data partitioning, fault tolerance, and performance optimization.
• Make system-level architectural decisions across resource allocation, execution models, batch versus streaming workloads, and consistency versus availability.
• Work with customers and AI-native companies to address challenges in scaling AI workloads.

Requirements

• 6+ years of experience building production-grade software, infrastructure, or developer-facing systems.
• Strong Python engineering experience.
• 6+ years personally owning core architectural decisions within a distributed data or compute engine.
• Deep experience with distributed-systems internals, including scheduling, fault tolerance, data partitioning, distributed execution, performance optimization, or database and query-engine internals.
• Demonstrated ability to reason through system-level tradeoffs and defend architectural decisions.
• Passion for large-scale AI infrastructure.

Compensation and Benefits

• $240,000–$270,000 annual salary.
• Equity and health, dental, and vision coverage, with many plans up to 99% employer-covered.
• Flexible time off, paid parental leave, and mental health support.