Software Engineer, Hardware
OpenAI
4 months ago
Remote, United States or San Francisco, CA, USA
Mid Level / Senior
Base Salary
$266k - $455k/yr
Responsibilities
- Design and build APIs and runtime components for heterogeneous ML workloads.
- Contribute to compiler infrastructure with optimizations and compiler passes.
- Engineer and optimize compute and data kernels for performance and portability.
- Profile and optimize system bottlenecks related to I/O and memory hierarchy.
- Develop simulation infrastructure for validating runtime behaviors and training stack changes.
- Rapidly deploy runtime and compiler updates in collaboration with hardware teams.
- Work across a diverse stack using Rust and Python.
Requirements
- Proficient in systems programming languages like Rust and C++.
- Experience with scripting languages such as Python.
- Familiarity with compiler development, kernel authoring, or runtime systems.
- Interest in large-scale systems and improving their performance and reliability.
- Ability to thrive in a fast-paced, collaborative environment.
Benefits
- Hybrid work model with 3 days/week in-office.
- Relocation assistance available.
Tech Stack
C++PythonRust
Categories
AI & MLBackend