Staff Software Engineer, SIMD Kernels - Santa Clara, United States

Originally posted 15 May 2026 by the employer.

About the role

As a member of the SIMD Kernels team, you will productize the software stack for the d-Matrix AI compute engine. You will develop, enhance, and maintain software kernels for ML operators across the hardware-software co-design toolchain.

What you'll do

- Develop, enhance, and maintain software kernels for ML operators, including softmax, layer norm, and activation functions. - Build solutions to make the SDK intuitive for developers to use and analyze performance. - Map algorithms and AI-framework computational graphs onto modern hardware architectures. - Navigate hardware-software co-design trade-offs across the full toolchain.

What you'll need

- MS or PhD in computer engineering, math, physics, or a related degree with 5+ years of industry experience. - Strong grasp of computer architecture, data structures, system software, and machine learning fundamentals. - Proficiency in C/C++ and Python development in a Linux environment using standard development tools. - Experience implementing algorithms for specialized hardware such as FPGAs, DSPs, GPUs, or AI accelerators using libraries such as CUDA. - Experience implementing operators commonly used in ML workloads, such as GEMMs, Convolutions, softmax, layer normalization, and pooling.

Nice to have

- Prior startup, small team, or incubation experience. - Experience with ML frameworks such as TensorFlow and PyTorch. - Experience working with ML compilers and algorithms, such as MLIR, LLVM, TVM, or Glow. - Experience with deep learning frameworks and ML models for CV, NLP, or recommendation. - Experience developing for embedded SIMD vector processors such as Tensilica. - Work experience at a cloud provider or AI compute/subsystem company.

Skills: SIMD Kernels, AI compute engine, ML operators, AI hardware, hardware-software co-design