Remote-Eligible Staff SIMD Kernel Engineer for AI Compute

d-Matrix; Santa Clara (CA); Hybrid; USD 120,000 - 150,000; Full time

Job summary

A cutting-edge tech company in Santa Clara is seeking a Software Engineer to develop software kernels for AI computation. The ideal candidate will have an MS or PhD, proficient skills in C/C++ and Python, and experience with ML algorithms. This role involves collaborating to improve the SDK for developers. A strong understanding of computer architecture and machine learning fundamentals is essential. Remote work options are available.

Qualifications

• 5+ years of industry experience in software development.
• Strong grasp of computational graphs in machine learning.
• Comfortable with hardware-software co-design trade-offs.

Responsibilities

• Develop, enhance, and maintain software kernels for ML operators.
• Build solutions for SDK usability and performance analysis.

Skills

C/C++ programming

Python development

Computer architecture

Machine learning fundamentals

Linux development

Algorithm implementation

Education

MS or PhD in computer engineering, math, physics, or related

Tools

CUDA

TensorFlow

PyTorch

Job description

A cutting-edge tech company in Santa Clara is seeking a Software Engineer to develop software kernels for AI computation. The ideal candidate will have an MS or PhD, proficient skills in C/C++ and Python, and experience with ML algorithms. This role involves collaborating to improve the SDK for developers. A strong understanding of computer architecture and machine learning fundamentals is essential. Remote work options are available.