Start Your Search Here

Job Search

Cerebras

Toronto / Global

Core ML Engineer at Cerebras Systems

Job Description

Elevate machine learning solutions with Cerebras Systems as a Core ML Engineer. Focus on runtime capabilities and efficient kernel development to optimize cutting-edge AI algorithms.

As part of the innovative Core ML team, you will bridge research ideas with high-performance execution on Cerebras' unique Wafer-Scale Engine. Your role requires a robust background in developing high-performance systems software, strong proficiency in C++ and Python, and a solid grasp of parallel programming concepts. Engage in translating research prototypes into efficient implementations that drive advanced AI workloads.

Key Responsibilities:

• Design runtime components for Core ML algorithms

• Implement novel high-performance kernels for optimized performance

• Profile and debug across layers of ML frameworks

• Optimize memory, communication, and concurrency

• Develop benchmarks and automated tests for validation

Requirements:

• Bachelor's, Master's, or PhD in a related field

• Experience with ML systems, runtimes, or computational kernels

• Strong programming in C++ and Python

• Debugging expertise across multiple software layers

• Familiarity with PyTorch or JAX

Transform research into reliable software with your skills at Cerebras Systems.

#J-18808-Ljbffr

Apply Now

Similar Opportunities

View all jobs