AI Kernel Writer

Company : Majestic Labs

Job Type : Full Time

Los Altos, United States

Number of Applicants

000+

Apply Now

Let AI Supercharge Your Job Hunt!

JobCopilot scans 500,000+ company career sites daily to find jobs for you

Never miss an opportunity Save hours by auto-filling applications forms Land more interviews with tailored applications

Activate JobCopilot

Job Description - AI Kernel Writer

Design and implement high-performance compute kernels for AI primitives such as GEMM, attention, normalization, and convolution.
Optimize for throughput, latency, and memory hierarchy across heterogeneous compute units (SIMD, matrix engines, DMA).
Collaborate with compiler and runtime teams to integrate kernels into Triton, PyTorch, or SYCL pipelines.
Profile and tune kernels using tools like Perfetto, VTune, Tracy, or custom simulators.
Prototype and evaluate precision formats (FP16/BF16/FP8/e5m2, etc.) and stochastic rounding.
Contribute to micro-architecture feedback loops, helping co-design ISA and memory features with the hardware team.
Write clear, well-structured, and reusable code (C++/CUDA/Triton/LLVM MLIR).

Bachelor's or Master's in Computer Science, Computer Engineering, or a related field from a recognized university.
Strong background in parallel programming (CUDA, Triton, SYCL, OpenCL, Metal, POSIX Threads, or OpenMP).
Experience with optimization of irregular algorithms, such as graph computations or sparse numerical linear algebra, combining high-level data structure design with low-level SIMD and synchronization optimizations.
Deep understanding of memory layout, vectorization, thread/block scheduling, and cache behavior.
Proficiency in C++11 or higher, with strong knowledge of standard algorithms, data structures, and generic programming paradigms.
Experience with code generation for high-performance computations and knowledge of frameworks like BLAS/BLIS/Torch
Skilled in performance analysis and parallel debugging using tools such as Valgrind, GNU Debugger, or CI testing frameworks.
Hands-on experience profiling and optimizing compute or AI workloads (e.g., GEMM, softmax, attention).
Solid grasp of numerical stability, precision formats, and mixed precision arithmetic.
Collaborative work style with the ability to operate effectively in multicultural, cross-disciplinary environments.

Original job AI Kernel Writer posted on GrabJobs ©. To flag any issues with this job please use the Report Job button on GrabJobs.

Apply Now

Auto-Apply to Similar Jobs

Share Job

Get your Resume Reviewed for Free

Automate Job Applications for Similar Jobs

About the Company

Majestic Labs

We are seeking a highly skilled Senior Power Engineer to take a leadership role in defining the board-level and rack-level power delivery architecture for our next-generation, multi-kilowatt AI server platforms. This critical position owns the end-to-end design of high-current power delivery paths t...

Similar AI Kernel Writer Jobs in the US

Get your Resume Reviewed for Free

Email address

Why are you reporting this job?

I think it’s a discriminatory or offensive

I think it’s fraudulent or a scam

I think it’s trying to sell something unrelated to the job / it’s asking for money

I think it contains incorrect or broken information

Other

All Job Ads are subject to GrabJobs’s Terms of Service. We allow users to flag postings that may be in violation of those terms. Job Ads may also be flagged by GrabJobs moderation team. However, no moderation system is perfect, and flagging a posting does not ensure that it will be removed.

Setup your job alert:

Frequency

By activating job alerts, I agree to GrabJobs Terms & Privacy Policy. I can unsubscribe to job alerts anytime. Skip

AI Kernel Writer

Job Description - AI Kernel Writer

About the Company

Similar AI Kernel Writer Jobs in the US

Mobile Apps