AI Performance Library Architect
Hillsboro, California, United States of America
Full Time
3 weeks ago
$170,500 - $315,490 USD
Visa Sponsor
Key skills
AssemblyLinuxPyTorchC++CAI
About this role
Role Overview
- design, development, and maintenance of new functionality in oneDNN to enable performance critical portions of AI workloads
- support software developers optimizing AI frameworks and workloads for Intel CPUs and GPUs, as well as cross-platform ecosystem of AI software developers contributing to oneDNN
Requirements
- Master’s degree in Mathematics, Physics, Computer Science, or a relevant STEM field
- OR Ph.D. degree in Mathematics, Physics, Computer Science, or a relevant STEM field
- 5+ years of experience in C and C++
- Maintaining or contributing to open-source software projects
- Software libraries design and architecture
- Implementation of linear algebra algorithms (functions from BLAS, LAPACK, or PyTorch)
- Performance engineering and software performance optimizations
- Floating point arithmetic and numerical stability
- Software development on Linux
- Low-level performance optimizations using CUDA, x86 assembly or intrinsics, or OpenCL
Tech Stack
- Assembly
- Linux
- PyTorch
Benefits
- competitive pay
- stock bonuses
- benefit programs which include health, retirement, and vacation