Sr. Principal Software Engineer, ML Compiler Front-End (AI2139)
Description
Sr. Principal Software Engineer, ML Compiler Front-End
Job ID- AI2139
Location: San Jose, CA (we are also open to remote work from within North America).
Job Description
SiMa.ai is looking for a Sr. Principal Software Engineer to join our world class team and lead the development of our state-of-the-art ML Compiler Front-End. As the Technical Lead of our ML Compiler Front-End, you will get to work with leading-edge ML technologies. You will also work with a talented team of engineers to optimize ML models so that they run efficiently on our Machine Learning System-on-a-Chip (MLSoc).
You will also have the opportunity to influence SiMa’s roadmap by proposing enhancements to both our software and future generations of the hardware.
Responsibilities
- Serve as the Technical Lead of the ML Compiler Front-End team, collaborating cross-functionally with other Technical Leads.
- Lead the analysis of ML applications and deep learning models to better understand their performance on the SiMa MLSoC, and further optimize them for performance.
- Lead efforts to enhance and support Front-End optimizations such as graph optimization, pruning, and quantization.
- Lead efforts to participate in and contribute to the TVM open source community.
- Lead efforts to improve the quality of the code base, such as code refactoring and documentation enhancement.
- Work closely with Program Management and Engineering Management to create project plans for the team’s activities, and provide regular status updates on how the team is progressing against its goals.
- Contribute to SiMa’s roadmap by proposing new features for both our software products and future generations of the MLSoC.
Required Background
- Ph.D. or M.S. in Computer Science or a related field with 10+ years of experience developing highly performant systems software.
- Strong communication and leadership skills.
- Strong programming skills in C, C++ or Python.
- Strong background in algorithms and data structures.
- Strong understanding of machine learning algorithms such as matrix multiplication and convolution.
- Experience with deep learning frameworks such as TensorFlow, PyTorch and ONNX.
- Good understanding of processor architecture.
- Highly desirable: prior experience with TVM and deep learning quantization schemes.
Personal attributes:
Can-do attitude. Strong team player. Curious, creative and good at solving problems. Execution and results-oriented. Self-driven, Thinks Big and is highly accountable. Good communication skills.