4 months ago
Base Salary
$180k - $300k/yr
Responsibilities
- Develop and maintain software kernels for AI hardware.
- Enhance the software stack for the AI compute engine.
- Collaborate with compiler experts and other software and hardware teams.
- Optimize hardware-software co-design for performance and efficiency.
- Implement algorithms for specialized hardware like FPGAs and GPUs.
Requirements
- MS in computer engineering, math, physics, or related field with 5+ years of experience, or PhD with 1+ years.
- Strong understanding of computer architecture and machine learning fundamentals.
- Proficient in C/C++ and Python in Linux environments.
- Experience with algorithms for specialized hardware using libraries like CUDA.
- Familiarity with ML frameworks such as TensorFlow or PyTorch is preferred.