Skip to content
New Job?Nejo!

The AI Job Search Engine

MA
Magic
1h ago

Kernel Engineer(m/w/x)

Wien
from USD 550,000 / year
Full-timeOn-site
AI/ML

Description

You will drive the company's performance by designing robust GPU kernels that solve complex memory challenges, ensuring seamless high-speed operations across training and inference systems.

Let AI find the perfect jobs for you!

Upload your CV and Nejo AI will find matching job offers for you.

Requirements

  • Deep expertise in computer architecture
  • Low-level machine optimizations expertise
  • Code generation expertise
  • Breadth across machine learning
  • Low-level programming for AI accelerators
  • GPU kernel development and optimization
  • Experience in kernel authoring frameworks
  • Agility, ownership mindset, and grit
  • Quick learning and grit

Tasks

  • Design and implement high-performance GPU kernels
  • Optimize kernels for training and inference throughput
  • Manage kernel design, deployment, and production reliability
  • Ensure robustness through extensive testing and functional correctness
  • Minimize latency and memory utilization for long-context windows
  • Evaluate porting compute kernels to alternative hardware
  • Collaborate with training, inference, and RL teams on co-design

Tools & Technologies

NVIDIA BlackwellGoogle TPUsNCCLMSCCLPPCUTLASSCuTeDSLTritonQuackFlash-AttentionPallasMosaicMojo

Languages

EnglishBusiness Fluent

Benefits

Competitive Pay

  • Equity compensation
  • 6% salary matching

Retirement Plans

  • 401(k) plan

Healthcare & Fitness

  • Health insurance
  • Dental insurance
  • Vision insurance

Family Support

  • Dependent insurance coverage

More Vacation Days

  • Unlimited paid time off

Additional Allowances

  • Relocation stipend
Find the original job posting in its most current version here. Nejo automatically captured this job from the website of Magic and processed the information on Nejo with the help of AI for you. Despite careful analysis, some information may be incomplete or inaccurate. Please always verify all details in the original posting! Content and copyrights of the original posting belong to the advertising company.
Not a perfect match?
100+ Similar Jobs in Wien
  • Magic

    Software Engineer - Supercomputing Platform & Infrastructure(m/w/x)

    Full-timeOn-siteExperienced
    from USD 550,000 / year
    Wien
  • Snap B.V.

    Embedded Engineer, Graphics(m/w/x)

    Full-timeOn-siteSenior
    Wien
  • Snap Inc

    Embedded Engineer(m/w/x)

    Full-timeOn-siteExperienced
    Wien
  • Snap Inc

    Embedded Engineer, Operating System(m/w/x)

    Full-timeOn-siteExperienced
    Wien
  • Snap Inc

    Machine Learning Engineer, Infrastructure(m/w/x)

    Full-timeOn-siteExperienced
    Wien
100+ View all similar jobs