Principal Software Engineer (Kernels)

Join our team as a Principal Software Engineer (Kernels) and be a key player in productizing the software stack for our AI compute engine. You will be responsible for the development, enhancement, and maintenance of software kernels for next-generation AI hardware. Collaborate with compiler experts and work closely with other software and hardware experts in the company. Enjoy a range of benefits including equity, health care, flexible time-off, and a remote/hybrid working model.

• Participer à l'équipe qui aide à productiser la pile logicielle pour notre moteur de calcul AI.
• Responsable du développement, de l'amélioration et de la maintenance des noyaux logiciels pour le matériel AI de prochaine génération.
• Travailler avec une équipe d'experts en compilateurs pour développer l'infrastructure du compilateur.
• You possess experience building software kernels for HW architectures
• You possess a very strong understanding of various hardware architectures and how to map algorithms to the architecture
• You understand how to map computational graphs generated by AI frameworks to the underlying architecture
• You have had past experience working across all aspects of the full-stack toolchain and understand the nuances of what it takes to optimize and trade off various aspects of hardware-software co-design
• You can build and scale software deliverables in a tight development window
• Experience implementing algorithms for specialized hardware such as FPGAs, DSPs, GPUs, and AI accelerators using libraries such as CUDA, etc
• Experience implementing algorithms in high-level languages such as C/C++ and Python
• Self-motivated team player with a strong sense of ownership and leadership
• Proficient in C/C++ and Python development in Linux environments and using standard development tools
• Strong grasp of computer architecture, data structures, system software, and machine learning fundamentals
• MS in computer engineering, math, physics, or a related degree with 12+ years of industry experience or a PhD in computer engineering, math, physics, or a related degree with 7+ years of industry experience
• Experience with development for embedded SIMD vector processors such as Tensilica
• Experience in implementing operators commonly used in ML workloads—GEMMs, Convolutions, BLAS, SIMD operators for operations like softmax, layer normalization, pooling, etc
• Prior startup, small team, or incubation experience
• Experience with ML frameworks such as TensorFlow and/orPyTorch
• Experience working with ML compilers and algorithms, such as MLIR, LLVM, TVM, Glow, etc
• Experience with a deep learning framework (such as PyTorch or TensorFlow) and ML models for CV, NLP, or recommendation
• Work experience at a cloud provider or AI compute/subsystem company