Software Engineer, Staff - SIMD Kernels
New
D
d-MatrixAI Hardware Software
Santa Clara, CA headquarter or any of our regional offices. Remote possible.Full-TimeStaff
Salary$190K - $300K; $190K – $300K • Offers Equity • Offers Bonus
Apply NowOpens the employer's application page
Job Details
- Experience
- 5+ years of industry experience
- Required Skills
- PythonMachine LearningPyTorchC++TensorflowLinux
Requirements
- MS or PhD in computer engineering, math, physics, or a related degree.
- 5+ years of industry experience.
- Strong grasp of computer architecture, data structures, system software, and machine learning fundamentals.
- Proficiency in C/C++ and Python development in Linux environments.
- Experience implementing algorithms in C/C++ and Python.
- Experience implementing algorithms for specialized hardware such as FPGAs, DSPs, GPUs, or AI accelerators using libraries such as CUDA.
- Experience implementing operators commonly used in ML workloads (e.g., GEMMs, Convolutions, softmax, layer normalization, pooling).
- Self-motivated team player with a strong sense of ownership and leadership.
Responsibilities
- Productize the software stack for the AI compute engine.
- Develop, enhance, and maintain software kernels for ML operators including softmax, layer norm, and activation functions.
- Optimize kernels for next-generation AI hardware.
- Build solutions to ensure the SDK is intuitive for developer use.
- Analyze performance and navigate hardware-software co-design trade-offs.
- Deliver and scale high-quality software within a fast-moving development environment.
View Full Description & ApplyYou'll be redirected to the employer's site
