As Machine Learning Software Engineer, you will contribute to the development of Zero ASIC’s machine learning solutions. You will be a strong contributor on a small team focused on designing, fabricating, and testing Zero ASIC devices.
Architect, design, and implement compiler support for Zero ASIC’s neural network inference platform
Develop and maintain ML compiler infrastructure, including benchmarking tools and automation scripts
Design and implement compiler optimizations for ML models
Work with hardware architects to improve the efficiency of system via software/hardware co-design
Stay up-to-date with the latest research and industry developments in the field of ML compilers.
BS/MS or PhD in Computer Science or related field
3+ years of industry experience
Strong background in compiler theory, algorithms, and optimization techniques
Strong understanding of ML operators and their implementations
Familiarity with parallel computer architectures
Familiarity with machine learning frameworks such as TensorFlow, PyTorch, and ONNX
Knowledge of computer architectures used for neural network inference, and neural network performance characteristics is a plus
Experience in ML compilers and MLIR is a plus