Senior Neural Network Kernel Software Development Engineer
RemoteUnited States
Job Summary
Design, prototype, and execute low-level, adaptable C++ programs for various neural net operations to accelerate execution on custom hardware accelerators. Define and document configuration APIs for these kernels to the compiler team while developing comprehensive computation strategies spanning multichannel and multi-chip implementations. Collaborate with architects and compiler engineers to share performance optimization concepts and maximize computation and communication throughput. Work within agile teams to tackle novel challenges using cutting-edge technologies, building innovative systems from scratch. Contribute to revolutionizing chip and software technologies with global impact.
Required Qualifications
- Degree in Computer Science, Engineering, Math, Physics, or related field (preferably MS or PhD)
- Profound knowledge of modern C++, with a focus on code generation and low-level compute optimizations
- Familiarity with fundamental Neural Network operator algorithms - Convolutions, Transformers, RNNs
- Demonstrated capability to independently navigate challenging, well-defined problems
- Aptitude and interest in both high-level conceptual understanding and intricate technical details
- Enthusiasm for problem-solving within highly structured and restricted environments
Desired Qualifications
- Proficiency in Python
- Experience with other AI accelerator programming
- Strong mathematical aptitude
- Enjoyment of solving complex problems
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.