Majestic Labs AI Logo

Majestic Labs AI

AI Kernel Engineer

Posted 2 Days Ago
Be an Early Applicant
In-Office
Los Altos, CA, USA
Entry level
In-Office
Los Altos, CA, USA
Entry level
Design and optimize high-performance AI compute kernels for GEMM, attention, normalization, convolution, and related primitives. Improve throughput, latency, memory behavior, and precision across heterogeneous hardware. Integrate kernels with Triton and PyTorch, profile workloads using performance tools, evaluate mixed-precision formats, and collaborate with compiler, runtime, and hardware teams on ISA and memory architecture co-design.
The summary above was generated by AI
Description


We’re a fast-moving, US-Israeli AI startup building next-generation infrastructure for the world’s most demanding AI workloads. Our mission is to accelerate the future of intelligence and make it ubiquitously accessible by delivering full stack custom AI servers optimized for large-scale inference and training.

Backed by top-tier investors and led by industry veterans, we’re scaling rapidly. We seek forward-thinking engineers and operators who thrive in a collaborative environment and excel at solving complex challenges from first principles. If you want to build infrastructure that fundamentally changes what the world can accomplish with AI, come join us!


About the position



  • 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 and PyTorch pipelines.
  • Profile and tune kernels using tools like Perfetto, VTune, Tracy, Nsight, or custom simulators.
  • Prototype and evaluate precision formats (FP16/BF16/FP8/e5m2, MXFP/FP4, etc.).
  • Contribute to micro-architecture feedback loops, helping co-design ISA and memory features with the hardware team.
Requirements


  • Degree at any level in Computer Science, Computer Engineering, or a related field from a recognized university.
  • Strong background in parallel programming (CUDA, Triton, RISC-V Vector (RVV), POSIX Threads, or OpenMP).
  • Deep understanding of memory layout, vectorization, thread/block scheduling, and cache behavior.
  • Experience programming wide-SIMD vector units and matrix/systolic engines.
  • Familiarity with the parallel-computing ecosystem and its high-performance libraries and
  • primitives, such as BLAS/BLIS, dense linear algebra, and parallel scan/reduce/sort.
  • Skilled in performance analysis and parallel debugging using tools such as GNU Debugger and Nsight.
  • 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.
  • Proficiency in C++17 or higher, with strong knowledge of standard algorithms, data structures, and generic programming paradigms.
  • Collaborative work style with the ability to operate effectively in multicultural, cross-disciplinary environments.

Preferred qualifications



  • 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.
  • Familiarity with LLVM/MLIR.
  • RISC-V systems or bare-metal programming, including RISC-V vector and matrix extensions.


We seek smart, curious, self-driven individuals who enjoy creative problem-solving and continuous learning. Even if you don't meet all requirements, we'd love to hear from you if you have the mindset to tackle complex challenges.

Majestic Labs is an equal opportunity employer deeply committed to diversity of background and thought. If you are ready to dare greatly, learn from mistakes, engage in vigorous debates, and help create a step-function shift in human technical capability, we want to meet you!

Similar Jobs

2 Months Ago
Hybrid
San Francisco, CA, USA
266K-445K Annually
Mid level
266K-445K Annually
Mid level
Artificial Intelligence • Machine Learning • Generative AI
The role involves enhancing kernel performance and AI-assisted tooling through developer workflows, observability infrastructure, and optimization techniques, collaborating across engineering teams.
Top Skills: AICompilersDeveloper ToolingKernel OptimizationObservabilityProfilingProgramming LanguagesReinforcement Learning
2 Months Ago
In-Office
San Francisco, CA, USA
Mid level
Mid level
Information Technology • Software
As a Research Engineer, you will optimize AI performance, focusing on kernel development for large-scale ML workloads, profiling bottlenecks, and collaborating with teams to enhance model training and inference.
Top Skills: AmdArmAws TrainiumCudaGoogle TpuHipHpcIntelPtxQualcommTriton
3 Minutes Ago
In-Office
Milpitas, CA, USA
99K-136K Annually
Mid level
99K-136K Annually
Mid level
Hardware • Other • Appliances
Develops, implements, and optimizes manufacturing processes for cooling products. Supports NPI, DFM, process transfers, root cause analysis, process validation, yield improvement, tooling and equipment implementation, supplier qualification, engineering changes, documentation, and production troubleshooting. Collaborates with engineering teams, suppliers, and production personnel to improve quality, cost, delivery, manufacturability, and productivity.
Top Skills: 3D CadAssembly AutomationLeak TestingSolidworksStatistical Process ControlTooling And Fixture Development

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account