Senior Machine Learning Compiler Engineer – C/C++
| Company | PlusAI |
|---|---|
| Location | Santa Clara, CA, USA |
| Salary | $170000 – $200000 |
| Type | Full-Time |
| Degrees | Bachelor’s, Master’s, PhD |
| Experience Level | Senior |
Requirements
- BS/MS/PhD in Computer Science, Computer Engineering, Electrical Engineering or related fields
- Proficient in C/C++
- 3+ years of industry experience
- Experience with AI inference computing optimization, quantization, model pruning techniques
- Experience with one or more of GPU/TPU/CPU/ML accelerators
- Experience with one or more of Machine Learning Compiler e.g. Tensorrt, TVM, XLA, Glow, or MLIR
Responsibilities
- Deploy, optimize and accelerate machine learning models on various computing platforms from e.g. Nvidia, TI, Ambarella etc
- Implement compiler support to maximize inference performance
- Ensure that your work is performed in accordance with the company’s Quality Management System (QMS) requirements and contribute to continuous improvement efforts.
- Ensure that technical work meets customer requirements, regulatory standards, and company quality policies.
Preferred Qualifications
- ML applications and ML optimization experience
- Familiar with one or more of the machine learning frameworks e.g. Pytorch/Tensorflow/AITemplate/JAX
- ML experience with large transformer models
- High performance computing experience
- Understand the principles of operations for sensors such as camera, radar and lidar