Software Engineer, Kernel Programming

ABOUT THE JOB Lead the integration of diverse AI models including VLA, Vision, and Multimodal architectures by utilizing our kernel programming language to ensure both accuracy and performance while keeping the stack ready for developers to use. RESPONSIBILITIES - Design and implement efficient kernels on FuriosaAI’s kernel programming stack (including vISA, TCL), targeting Tensor Contract Processor (TCP) architectures. - Diagnose and optimize kernel performance with profiling tools and roofline analysis for each RNGD-accelerated AI model. - Develop and apply automated kernel generation and optimization for AI workloads. - Build diagnostic tools or testbeds for robust and reliable kernel validation. - Drive end-to-end programming enablement on RNGDs, creating reproducible guides and reference implementations. MINIMUM QUALIFICATIONS - BS in Computer Science, Artificial Intelligence, Electrical Engineering, or a related field. - Experience in low-level systems programming targeting XPU (e.g., NPU, GPU) architectures. - Experience collaborating across engineering, research, and product teams to align software development with product requirements. PREFERRED QUALIFICATIONS - MS or PhD in Computer Science, Artificial Intelligence, Electrical Engineering, or a related field. - Experience in optimizing high-performance kernels on AI accelerators (e.g., GPU, TPU) for AI products. - Understanding of XPU architecture (computation patterns, data movement) and software-hardware co-optimization strategies. - Experience in open-source or research projects on AI model architectures such as Diffusion, Mamba, and VLA. - Experience in designing efficient deep learning architectures and developing algorithms for AI applications. CONTACT - recruit@furiosa.ai