Senior Staff Software Engineer - Kernels

Stack

Responsibilities

• ▹Develop, enhance and maintain software kernels for next-generation AI hardware
• ▹Map computational graphs generated by AI frameworks to the underlying architecture
• ▹Work across the full-stack toolchain and optimize the trade-offs of hardware-software co-design
• ▹Build and scale software deliverables in a tight development window
• ▹Work with a team of compiler experts on the compiler infrastructure, closely with ML, systems and hardware (mixed signal, DSP, CPU) experts

Requirements

• ▹MS in computer engineering, math, physics or a related field with 10+ years of industry experience, or a PhD with 1+ years of industry experience
• ▹Strong grasp of computer architecture, data structures, system software and machine learning fundamentals
• ▹Proficient in C/C++ and Python development in Linux environments with standard development tools
• ▹Experience implementing algorithms in high-level languages such as C/C++ and Python
• ▹Experience implementing algorithms for specialized hardware such as FPGAs, DSPs, GPUs and AI accelerators, using libraries such as CUDA
• ▹Experience implementing operators common in ML workloads: GEMMs, convolutions, BLAS, SIMD operators for softmax, layer normalization, pooling and more
• ▹Experience with development for embedded SIMD vector processors such as Tensilica
• ▹Self-motivated team player with a strong sense of ownership and leadership

Nice to have

• ▹Prior startup, small team or incubation experience
• ▹Experience with ML frameworks such as TensorFlow and/or PyTorch
• ▹Experience with ML compilers and algorithms such as MLIR, LLVM, TVM, Glow
• ▹Experience with a deep learning framework and ML models for CV, NLP or recommendation
• ▹Work experience at a cloud provider or AI compute/subsystem company

Soft skills

About the company

d-Matrix is focused on unleashing the potential of generative AI to power the transformation of technology, at the forefront of software and hardware innovation. Its culture is built on respect, collaboration, humility and direct communication.