Platform Workload Enablement Engineer

OpenAI; Seattle, San Francisco (WA, CA); Hybrid; USD 120,000 - 150,000; Full time

Benefits offered by this job

Equal Opportunity Employer

Commitment to diversity and inclusion

Reasonable accommodations available

Job summary

A leading AI research company in Seattle is seeking a Software Engineer to enhance production workflows and enable testing on new platforms. Responsibilities include creating test harnesses, validating workloads, and ensuring performance across various systems. The ideal candidate will hold a BS in CS/EE with over 5 years' experience in ML systems, distributed systems, and advanced profiling skills. This position offers the opportunity to work with cutting-edge technology in a dynamic and innovative environment.

Qualifications

• 5+ years in ML systems or performance engineering.
• Experience with large-scale distributed training concepts.
• Ability to analyze performance bottlenecks.

Responsibilities

• Port and validate key inference and training workloads.
• Build benchmarks and stress tests for workloads.
• Create repeatable test harnesses running in CI environments.

Skills

Performance engineering

Distributed systems

Hands-on experience with PyTorch

Profiling/debugging skills

Education

BS in CS/EE or equivalent practical experience

Tools

C++

CUDA

Python

Job description

A leading AI research company in Seattle is seeking a Software Engineer to enhance production workflows and enable testing on new platforms. Responsibilities include creating test harnesses, validating workloads, and ensuring performance across various systems. The ideal candidate will hold a BS in CS/EE with over 5 years' experience in ML systems, distributed systems, and advanced profiling skills. This position offers the opportunity to work with cutting-edge technology in a dynamic and innovative environment.