Software Engineer, Systems ML - Compilers / Kernels

Job Description

Develop and optimize AI compiler, framework, runtime, and high-performance kernel components for current and next-generation MTIA hardware. Analyze deep learning models and collaborate with AI researchers and hardware teams to improve training and inference performance.

Requirements: Requires proven C/C++ programming skills and experience developing AI frameworks or accelerating deep learning models on hardware. Preferred qualifications include relevant experience with AI frameworks, performance optimization, accelerator architectures, and kernels, plus a qualifying degree and experience level: a bachelor's degree with 7+ years, a master's degree with 4+ years, or a PhD with 3+ years.

Key Skills: C/C++, AI Compiler Development, PyTorch, Deep Learning, High-Performance Kernel Development, Compiler Optimization, Performance Tuning, AI Framework Development, CUDA, OpenMP, OpenCL, AI Hardware Accelerators, GPU Architecture, Runtime Analysis, Responsible AI

Benefits: Bonus, Equity, Benefits