Senior Python/PyTorch Inference Engineer Frontend/Compiler

Cerebras Systems, Inc.; Dubai; On-site; AED 367,242 - 514,138; Full time

Benefits offered by this job

Equal opportunity employer

Simple, non-corporate work culture

Support for continuous learning and growth

Job summary

Cerebras Systems, Inc. is seeking a Senior Software Engineer for the Inference ML Engineering team. The role involves designing APIs and ML features for generative AI models on custom hardware, collaborating across teams to build scalable inference solutions.

The ideal candidate will have extensive experience in software engineering, particularly in deep learning, and skills in Python and C++. Join us to work on cutting-edge AI technology in an inclusive and dynamic environment.

Qualifications

• 8+ years of experience in large-scale software engineering, focusing on deep learning.
• Proficiency in Python for scalable systems.
• Advanced proficiency in C++ with emphasis on multi-threaded programming.

Responsibilities

• Drive technical guidance to a team on complex machine learning integration.
• Design and implement ML features enhancing generative AI models.
• Maintain scalable serving backend for concurrent requests.

Skills

Python

C++

Deep Learning

Machine Learning Frameworks (PyTorch)

Problem Solving

Education

Bachelor's, Master's, or PhD in Computer Science, Computer Engineering, Mathematics, or related field

Tools

vLLM

TensorRT-LLM

Job description

Cerebras Systems, Inc. is seeking a Senior Software Engineer for the Inference ML Engineering team. The role involves designing APIs and ML features for generative AI models on custom hardware, collaborating across teams to build scalable inference solutions.

The ideal candidate will have extensive experience in software engineering, particularly in deep learning, and skills in Python and C++. Join us to work on cutting-edge AI technology in an inclusive and dynamic environment.