Senior Staff Software Engineer - Kernels
Belgrade • FullTime
Posted 6mo ago
About the job
At d-Matrix, we are at the forefront of software and hardware innovation, focused on unleashing the potential of generative AI. We are seeking passionate individuals driven by execution to join our inclusive and collaborative team. Together, we can shape the endless possibilities of AI.
Responsibilities
- Develop, enhance, and maintain software kernels for next-generation AI hardware.
- Build software kernels for HW architectures, mapping algorithms to the architecture.
- Map computational graphs from AI frameworks to the underlying architecture.
- Optimize hardware-software co-design aspects.
- Build and scale software deliverables within tight development windows.
- Collaborate with compiler experts to build compiler infrastructure.
- Work closely with software (ML and systems) and hardware experts.
Requirements
- MS in computer engineering, math, physics, or related with 10+ years of industry experience, OR PhD with 1+ years of industry experience.
- Strong understanding of computer architecture, data structures, system software, and machine learning fundamentals.
- Proficient in C/C++ and Python development in Linux environments.
- 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 ML workload operators (GEMMs, Convolutions, BLAS, SIMD operators).
- Experience with development for embedded SIMD vector processors (e.g., Tensilica).
- Self-motivated team player with ownership and leadership skills.
- Experience with ML frameworks such as TensorFlow and/or PyTorch (preferred).
- Experience with ML compilers and algorithms (MLIR, LLVM, TVM, Glow) (preferred).
- Experience with a deep learning framework and ML models for CV, NLP, or recommendation (preferred).
- Work experience at a cloud provider or AI compute/subsystem company (preferred).