4 months ago
Base Salary
$190k - $300k/yr
Responsibilities
- Develop and maintain software kernels for ML operators like softmax and layer normalization.
- Enhance the software stack for the AI compute engine.
- Build intuitive solutions for the SDK to aid developers.
- Analyze performance and optimize software for next-generation AI hardware.
- Collaborate with team members to navigate hardware-software co-design trade-offs.
Requirements
- MS or PhD in computer engineering, math, physics, or a related field with 5+ years of industry experience.
- Strong understanding of computer architecture, data structures, and machine learning fundamentals.
- Proficient in C/C++ and Python development in a Linux environment.
- Experience implementing algorithms for specialized hardware such as FPGAs and GPUs.
- Self-motivated team player with a strong sense of ownership and leadership.