Staff Software Engineer, SIMD Kernels
$190k - $300k • Remote • Santa Clara • FullTime
Posted 2mo ago
About the job
At d-Matrix, we are focused on unleashing the potential of generative AI to power the transformation of technology. We are at the forefront of software and hardware innovation, pushing the boundaries of what is possible. Our culture is one of respect and collaboration, valuing humility and direct communication. We are seeking individuals passionate about tackling challenges and driven by execution to help shape the endless possibilities of AI.
Responsibilities
- Productize the software stack for our AI compute engine.
- Develop, enhance, and maintain software kernels for ML operators (e.g., softmax, layer norm, activation functions) for next-generation AI hardware.
- Build solutions to make our SDK intuitive for developers and analyze performance.
Requirements
- MS or PhD in computer engineering, math, physics, or related field with 5+ 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 a Linux environment using standard development tools.
- Experience implementing algorithms in C/C++ and Python.
- Experience implementing algorithms for specialized hardware (FPGAs, DSPs, GPUs, AI accelerators) using libraries like CUDA.
- Experience implementing operators common in ML workloads (GEMMs, Convolutions, softmax, layer normalization, pooling).
- Self-motivated team player with strong ownership and leadership skills.
- Experience with ML frameworks such as TensorFlow and/or PyTorch.
- Experience with ML compilers and algorithms (MLIR, LLVM, TVM, Glow).
- Experience with a deep learning framework and ML models for CV, NLP, or recommendation.
- Experience in development for embedded SIMD vector processors (e.g., Tensilica).