Senior Software Engineer - Kernels
Belgrade • FullTime
Posted 4mo 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 foster a culture of respect, collaboration, humility, and direct communication, valuing diverse perspectives to create better solutions. We are seeking passionate individuals driven by execution to join our inclusive team and help 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 with ML, systems, and hardware experts.
Requirements
- MS in computer engineering, math, physics, or related with 5+ 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).