Senior AI Software Engineer

Job Summary

A company is looking for a Senior AI Software Engineer, Kernel Libraries.

Key Responsibilities

• Innovating and developing new AI systems technologies for efficient inference
• Designing, implementing, and optimizing kernels for high impact AI workloads
• Collaborating closely with other engineers across deep learning frameworks and GPU architecture teams

Required Qualifications

• Masters degree in Computer Science, Electrical Engineering, or related field (PhD preferred)
• 6+ years of experience with ML/DL systems development
• Strong experience with deep learning frameworks (e.g. PyTorch, JAX, TensorFlow)
• Proficient in Python and C/C++ programming
• Experience with inference engines and runtimes is advantageous