Compiler Software Development Engineer
Core
Develop a machine learning model compiler for AMD AIE in Ryzen AI PC and embedded FPGA devices, converting ONNX/PyTorch models into optimized code for VLIW processor arrays and handling partitioning between x86/embedded processors and AIE.
Role type
Senior IC compiler engineer (ML hardware accelerator)
Builds
Optimized control and executable code for AMD AIE VLIW processor arrays
Domain
Semiconductor / Machine Learning Infrastructure
Deliverable
production ML models
Required skills
C/C++, Python, MLIR, LLVM, compiler building, code generation for hardware accelerators, ONNX, PyTorch, concurrent programming, threading APIs
Preferred skills
Debugging, source code control systems, profilers, modern software development processes
Technologies
MLIR, LLVM, ONNX, PyTorch, GitHub
Responsibilities
Integrate compiler and compiled models with ML Frameworks; Implement model partitioning in ML Frameworks and/or MLIR; Implement runtime to distribute work to and collect results from x86 cores and AIE cores; Debug and fix existing issues and research alternative, more efficient ways to accomplish the same work
Seniority
Senior, hands-on IC
