PyTorch Frontend Inference Compiler Engineer

Cerebras Systems; United Arab Emirates; On-site; AED 150,000 - 200,000; Full time

Job summary

A leading AI technology firm in Dubai is seeking a Python / PyTorch Developer for frontend inference compiler development. You will analyze generative models and enhance compiler infrastructure to optimize performance on unique hardware. The ideal candidate has a degree in engineering or computer science and strong Python and PyTorch skills, with knowledge of compiler technology. This role offers an innovative environment focused on groundbreaking advancements in AI.

Benefits

Inclusive work culture

Job stability with startup vitality

Opportunity to publish and open source AI research

Qualifications

• Strong Python programming skills and in-depth experience with PyTorch internals.
• Solid understanding of computational graphs and tensor operations.
• Experience building or extending compilers or ML optimization frameworks.

Responsibilities

• Analysis of new models and impacts on compilation stack.
• Develop and maintain frontend compiler infrastructure for PyTorch models.
• Collaborate with other teams throughout feature implementation.

Skills

Python programming

PyTorch internals

Computational graphs understanding

Compiler experience

C++ extensions familiarity

ML model optimization

Education

Degree in Engineering or Computer Science

Tools

LLVM

MLIR

HuggingFace Transformers

Job description

A leading AI technology firm in Dubai is seeking a Python / PyTorch Developer for frontend inference compiler development. You will analyze generative models and enhance compiler infrastructure to optimize performance on unique hardware. The ideal candidate has a degree in engineering or computer science and strong Python and PyTorch skills, with knowledge of compiler technology. This role offers an innovative environment focused on groundbreaking advancements in AI.