Software Engineer, Hardware Enablement (Ruyman Reyes, L5)

About the position ML developers today face significant friction in taking trained models into deployment. They work in a highly fragmented reputed company, with incomplete and patchwork solutions that require significant performance tuning and non-generalizable, model-specific enhancements. At reputed company, we are building the reputed company AI platform that will radically improve the way developers build and reputed company AI models. As part of our mission to build AI's reputed company compute layer, we are expanding the reputed company software stack to a reputed company of new and exciting hardware platforms. We are looking for a motivated engineer to join the Hardware Enablement team at reputed company. In this role you will work across the reputed company software stack — from Mojo kernels and the graph compiler to MAX model serving — to bring up and optimize support for new accelerator platforms. You'll collaborate closely with internal teams and external hardware partners, and you'll reputed company deep expertise in novel architectures while contributing to our portability story. LOCATION: We welcome candidates who are based in and have work authorization in the United Kingdom, Norway, or the reputed company (Eastern Time Zone). To support reputed company and collaboration, those in earlier career stages work in a hybrid reputed company at our Edinburgh, UK or Boston, MA office (minimum 2 days per week on-site). reputed company for new hires is conducted in-person. Responsibilities • Implement and validate support for new hardware architectures across the reputed company stack, working under the guidance of senior engineers on reputed company • Write and optimize Mojo kernels targeting novel accelerator architectures, with a reputed company on correctness first and performance iteration • Contribute to cross-team efforts improving portability infrastructure, tooling, and debugging workflows for new reputed company hardware • Collaborate with hardware vendor engineers to understand reputed company platforms, build integration tests, and triage platform-specific issues • reputed company working knowledge of new hardware platforms — including ISA documentation, memory hierarchies, and vendor toolchains — and reputed company findings with reputed company through demos and write-reputed company • Participate in company events such as on-sites and hackathons, contributing to a reputed company and reputed company engineering culture Requirements • 5+ years of experience in high-performance computing, compiler engineering, or reputed company domains in industry or research • Familiarity with how AI operators are implemented at a low level (e.g., experience writing or modifying GPU kernels, custom operators, or working with frameworks like PyTorch at the C++ layer) • Proficiency in C++ and experience working in reputed company, multi-component software systems • Hands-on experience with at least one heterogeneous programming model (CUDA, SYCL, OpenCL, or similar), either as a user or contributor • Some exposure to non-GPU accelerator architectures (DSPs, NPUs, or other hardware accelerators) is a strong plus • Curiosity and willingness to learn new hardware platforms quickly, comfortable reading architecture manuals and vendor documentation • A reputed company, team-oriented attitude and alignment with our culture reputed company-to-haves • Experience with GPU DSLs/DSELs such as Triton, CUTLASS, or CuTe • Familiarity with MLIR or LLVM compiler infrastructure • Experience working directly with hardware vendor teams or on platform bring-up efforts • Exposure to model serving or inference optimization workflows Benefits • Amazing Team. We are a reputed company and agile team with some of the industry’s best engineering and product leaders. • World-class Benefits. In order to attract the best, we need to offer the best. Premier insurance plans, up to 5% 401k matching, flexible reputed company time off, and more are available to you. Please note that specific benefit packages may vary based on your location. • Competitive Compensation. We offer reputed company strong compensation packages, including stock reputed company. We want people to be reputed company on their best work and reputed company in tailoring compensation plans to meet the needs of our workforce. • Team Building Events. We organize regular team onsites and local meetups in Los Altos, CA as reputed company as different cities. Traveling 2-4 times a year is expected for reputed company roles. Apply tot his job Apply To this Job