Senior Kernel Optimization Engineer

Job not on LinkedIn

🕒 May 21

🏄 California – Remote

🏄 California – Remote

⏰ Full Time

🟠 Senior

👷🏻‍♀️ Engineer

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

Apply Now

Find Similar Remote Jobs

Receive Emails For Remote Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

Cerebras Systems

WebsiteLinkedInAll Job Openings

201 - 500 employees

Founded 2016

🤖 Artificial Intelligence

🔧 Hardware

⚕️ Healthcare Insurance

Artificial Intelligence • Hardware • Healthcare Insurance

Cerebras Systems is a pioneering company that focuses on developing advanced AI hardware, specifically the Cerebras Wafer Scale Engine, which delivers unparalleled performance in AI inference, outperforming traditional GPU setups. Their cutting-edge technology enables organizations like Mayo Clinic and AlphaSense to run state-of-the-art AI models with remarkable speed and efficiency. With flexible deployment options including cloud and on-premises solutions, Cerebras is transforming the landscape of AI capabilities for innovative teams across various industries.

📋 Description

• Develop design specifications for new machine learning and linear algebra kernels and mapping to the Cerebras WSE System using various parallel programming algorithms. • Develop and debug kernel library of highly optimized low level assembly instruction and C-like domain specific language routines to implement algorithms targeting the Cerebras hardware system. • Develop and debug high-performance kernel routines in low-level assembly and a custom C-like (CSL) language, implementing algorithms optimized for the Cerebras hardware system. • Using mathematical models and analysis to measure the software performance and inform design decisions. • Develop and integrate unit and system testing methodologies to verify correct functionality and performance of kernel libraries. • Study emerging trends in Machine Learning applications and help evolve Kernel library architecture to address computational challenges of the start-of-the-art Neural Networks. • Interact with chip and system architects to optimize instruction sets, microarchitecture, and IO of next generation systems.

🎯 Requirements

• Bachelor’s, Master’s, PhD or foreign equivalents in Computer Science, Computer Engineering, Mathematics, or related fields. • Understanding of hardware architecture concepts — must be comfortable learning the details of a new hardware architecture. • Skilled in C++ and Python programming languages. • Good knowledge of library and/or API development best practices. • Strong debugging skills and knowledge of debugging complex software stack. • Experience in kernel development and/or testing. • Familiarity with parallel algorithms and distributed memory systems. • Experience in programming accelerators such as GPUs and FPGAs. • Familiarity with Machine Learning neural networks and frameworks such as TensorFlow and PyTorch. • Familiarity with HPC kernels and their optimization.

🏖️ Benefits

• Build a breakthrough AI platform beyond the constraints of the GPU. • Publish and open source their cutting-edge AI research. • Work on one of the fastest AI supercomputers in the world. • Enjoy job stability with startup vitality. • Our simple, non-corporate work culture that respects individual beliefs.