We are seeking a talented engineer to work on our AI software stack—specifically, the low-level optimization of AI models and kernels for our unique hardware architecture.
Requirements
B.Sc. degree in software engineering, computer science, or a related field.
5+ years of experience in C/C++ programming with a focus on performance-critical, distributed, or resource-constrained systems.
Proficiency in writing AI kernels for GPUs such as GEMM and FlashAttention, and in low level software/firmware implementations
Good understanding of HW/SW interfaces, memory management, SIMD/vectorization
Excellent communication and teamwork skills
Strong problem-solving skills and the ability to analyze and debug complex issues.
Experience with ML accelerators and AI compilers - a plus.
Responsibilities
Develop our optimized AI libraries and related components in the SW stack.
Drive the performance optimization of AI models running on our unique architecture.
