Software Engineer, Hardware

Responsibilities

• Design and build APIs and runtime components to orchestrate computation and data movement across heterogeneous ML workloads.
• Contribute to compiler infrastructure, including the development of optimizations and compiler passes to support evolving hardware.
• Engineer and optimize compute and data kernels, ensuring correctness, high performance, and portability across simulation and production environments.
• Profile and optimize system bottlenecks, especially around I/O, memory hierarchy, and interconnects, at both local and distributed scales.
• Develop simulation infrastructure to validate runtime behaviors, test training stack changes, and support early-stage hardware and system development.
• Rapidly deploy runtime and compiler updates to new supercomputing builds in close collaboration with hardware and research teams.
• Work across a diverse stack, primarily using Rust and Python, with opportunities to influence architecture decisions across the training framework.

Requirements

• Deep curiosity for how large-scale systems work and enjoy making them faster, simpler, and more reliable.
• Proficient in systems programming (Rust, C++) and scripting languages like Python.
• Experience in one or more of the following areas: compiler development, kernel authoring, accelerator programming, runtime systems, distributed systems, or high-performance simulation.
• Excited to work in a fast-paced, highly collaborative environment with evolving hardware and ML system demands.
• Value engineering excellence, technical leadership, and thoughtful system design.