Machine Learning Compiler Engineer

Ceremorphic, Inc.; Hyderabad; On-site; INR 700,000 - 1,100,000; Full time

Job summary

Ceremorphic, Inc. is seeking a Deep Learning Compiler Engineer to build and optimize tools that compile state-of-the-art neural networks for Ceremorphic chips.

You will collaborate with DL software framework teams and hardware architecture groups to accelerate next-generation software, define APIs, and drive optimization techniques across the compiler stack.

Qualifications

• Strong C/C++ programming with debugging, performance analysis and test design.
• Experience with high level ML frameworks (TensorFlow, PyTorch, MXNet).
• Knowledge of ML hardware accelerators and common architectures (ResNet, MobileNet, SSD).
• Ability to work independently, define goals and lead development.
• Experience with deep learning compiler space (ONNC, TVM, XLA) is a plus.
• Knowledge of OpenCL programming is preferable.

Responsibilities

• Develop tools to compile state-of-the-art DL models for Ceremorphic chips.
• Collaborate with DL software framework teams and hardware architecture teams.
• Define public APIs, performance tuning and analysis for compiler optimizations.
• Craft and implement compiler/optimization techniques for neural networks.
• Contribute to general software engineering tasks and projects.

Skills

C/C++ programming

Software design

Debugging

Performance analysis

Test design

TensorFlow

PyTorch

MXNet

ONNC

TVM

XLA

OpenCL

CNN architectures

Leadership

Tools

OpenCL

Job description

We are looking for Deep Learning Compiler Engineers. In this role, you will be responsible for developing the tools used to compile state of the art deep learning models for custom Ceremorphic chips. You’ll collaborate with members of the deep learning software framework teams and the hardware architecture teams to accelerate the next generation of deep learning software. The scope of these efforts includes defining public APIs, performance tuning and analysis, crafting and implementing compiler and optimization techniques for neural networks, and other general software engineering work.

Key Requirements

• Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design.
• Experience working with high level machine learning frameworks (Tensorflow, PyTorch, MXNet)
• Knowledge of the machine learning hardware accelerator space (basic architectures, common techniques shared across the space, etc)
• Familiar with popular network architectures (ResNet, MobileNet, SSD, etc)
• Ability to work independently, define project goals and scope, and lead your own development effort
• Experience with the deep learning compiler space (ONNC, TVM, XLA, etc)is a huge plus
• Knowledge on OpenCL programming experience is preferable.
• Experience with the following technologies: deep learning models and algorithms, deep learning framework design, high performance compiler design.