Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Trainium Kernel Engineer - NKI AI Trainer

Mercor Inc

Mercor seeks an experienced kernel-evaluation engineer to assess NKI development tasks for frontier AI models. You will judge CUDA→NKI migration fidelity, Trainium-specific performance quality, and cross-platform numerical-correctness, delivering rubric-based feedback. Applicants should have 2+ years in NKI kernel work, with knowledge of tile-based computation, SBUF/PSUM/HBM memory management, and DMA orchestration, plus familiarity with AWS Neuron SDK and Neuron Compiler internals. #J-18808-Ljbffr Mercor Inc

Vacancy posted 4 days ago
Similar jobs that could be interesting for youBased on the Trainium Kernel Engineer - NKI AI Trainer in San Francisco, CA vacancy
  •  ...hardware-appropriateness of Neuron Kernel Interface (NKI) development tasks used to...  ...and evaluate a frontier AI lab's models. You will...  ...CUDA→NKI migration fidelity, Trainium-specific performance, cross-...  ..., you will work with kernel engineers and ML developers to ensure... 
    Suggested

    Mercor

    San Francisco, CA
    4 days ago
  • $70 - $90 per hour

     ...Mercor connects elite creative and technical talent with leading AI research labs. Headquartered in San Francisco, our investors...  ..., Larry Summers , and Jack Dorsey . Position: Trainium (NKI) Kernel Expert Type: Contract Compensation: $70–$90/hour... 
    Suggested
    Contract work
    Summer work
    Remote work

    Mercor

    San Francisco, CA
    4 days ago
  •  ...hardware-appropriateness of Neuron Kernel Interface (NKI) development tasks used to...  ...and evaluate a frontier AI lab's models. You'll assess...  ...CUDA→NKI migration fidelity, Trainium-specific performance-...  ...pipeline utilization, tensor-engine throughput, memory-bandwidth... 
    Suggested

    Obsidian

    San Francisco, CA
    4 days ago
  • OpenAI in San Francisco seeks an experienced Software Engineer to help bring inference workloads to AWS Trainium and build the software stack to run frontier...  ...platform. This deeply technical, cross‑stack role covers kernels, compilers, and model execution. You will develop... 
    Suggested

    Slope

    San Francisco, CA
    2 days ago
  • Join a leading AI lab's cutting-edge GenAI team to be at the core of the AI revolution, where your expertise...  ...from the ground up. We're seeking talented MLOps Engineers with deep, hands-on expertise in PyTorch and kernel-level programming (Triton/Pallas). This role... 
    Suggested
    Full time
    Weekday work

    Obsidian

    San Francisco, CA
    2 days ago
  • $190k - $250k

     ...Sciforium is an AI infrastructure company developing next-generation multimodal AI models...  ...from AMD with hands-on support from AMD engineers the team is scaling rapidly to build the...  ...We are seeking a highly skilled GPU Kernel Engineer who is passionate about pushing... 
    Full time
    Flexible hours

    Sciforium

    San Francisco, CA
    4 days ago
  • $40 per hour

    AI Trainer - Materials Science Specialists (Remote Advisory) We are seeking AI Trainers - Materials Science Specialists to join our expert B2B and AI research network. Our work supports global organizations with high‑level corporate intelligence, and we rely on executive... 
    Remote job

    Prolific

    San Francisco, CA
    5 days ago
  • $40 per hour

    A leading AI research firm is looking for AI Trainers with specialized knowledge in Materials Science. The role involves reviewing scientific papers and graphical abstracts to ensure accuracy while improving AI models. Candidates should have a Master's Degree or higher,... 
    Remote job
    Flexible hours

    Prolific

    San Francisco, CA
    5 days ago
  • Magic is hiring a Kernel Engineer in San Francisco to design, implement, and optimize high-performance kernels for long-context training and...  ...on robustness, extensive testing, and quality. Experience with AI accelerators and GPU kernel frameworks is essential; visa... 
    Visa sponsorship

    Magic

    San Francisco, CA
    1 day ago
  • Mercor is seeking GPU kernel optimization experts to contribute to a project with a leading AI lab. This contract-based opportunity is designed for freelancers with strong C++ skills, practical GPU programming experience, and the ability to squeeze performance out of modern... 
    Contract work
    Freelance

    Mercor

    San Francisco, CA
    2 days ago
  • Inception is seeking engineers and scientists to design, optimize, and maintain compute foundations for large‑scale language model training and inference. You will develop high‑performance ML kernels, enable efficient low‑precision arithmetic, and improve the distributed... 

    Inception

    San Francisco, CA
    5 days ago
  • Mercor is seeking GPU kernel optimization experts to contribute to a project with a leading AI lab. This contract-based opportunity is designed for freelancers with strong C++ skills, practical GPU programming experience, and the ability to improve kernel performance using... 
    Contract work
    Freelance

    Obsidian

    San Francisco, CA
    3 days ago
  • 1. Role Overview Mercor is seeking GPU kernel optimization experts to contribute to a project with a leading AI lab. This opportunity is designed for freelancers with strong C++ skills, practical GPU programming experience, and the ability to improve kernel performance... 
    Contract work
    Freelance

    Obsidian

    San Francisco, CA
    2 days ago
  •  ...software for inference and agent hosting. In this role, you’ll own token processing down to the lowest layers of the stack, optimize kernel performance, develop new request scheduling and parallelism strategies, and help us use a heterogeneous mix of hardware at max... 

    Sail

    San Francisco, CA
    4 days ago
  •  ...creative and technical talent with leading AI research labs. Headquartered in San...  ..., and Jack Dorsey . Position: CUDA Engineering Expert Type: Contract Compensation...  ...Responsibilities Analyze and optimize GPU kernels for performance, efficiency, and... 
    Contract work
    Summer work
    Remote work

    Mercor

    San Francisco, CA
    14 days ago
  • $30 per hour

     ...A leading platform for AI training is seeking advanced Japanese speakers to join as Domain Expert participants. You will complete various AI training tasks and evaluate AI responses in Japanese. Compensation is competitive at $30 per hour, with flexible working hours... 
    Hourly pay
    Remote work
    Work from home
    Flexible hours

    Prolific

    San Francisco, CA
    2 days ago
  •  ...Associates Limited is seeking a highly skilled engineer to design and build high-performance...  ...systems, working close to the Linux kernel, containers, and PCIe, with production-grade...  ...delivering scalable platforms used by leading AI labs. #J-18808-Ljbffr Hamilton Barnes... 
    Remote job

    Hamilton Barnes Associates Limited

    San Francisco, CA
    2 days ago
  • $106.9k - $200.6k

     ...career wherever you want it to go. Join EY and help to build a better working world. The opportunity We are seeking an AI Systems Engineer to own the stateful backbone of EY’s AI-native platform, including the data stores, memory tiers, and event streaming systems... 
    Full time
    Summer holiday
    Flexible hours

    EY

    San Francisco, CA
    5 days ago
  • $150 per hour

    A leading AI data collection platform is seeking AI Trainers - Machine Learning Specialists to assist in training and evaluating advanced AI models. Successful candidates will complete training tasks, provide expert feedback, and help enhance AI's capabilities. This role... 
    Remote job
    Work from home

    Prolific

    San Francisco, CA
    3 days ago
  • AI Systems Engineer - Codex Core Agents Location San Francisco Employment Type Full time Department Applied AI Compensation 230K-385K Offers...  ...or post‑training feedback loops. Background in compilers, kernels, runtimes, inference optimization, GPU systems, benchmarking... 
    Full time
    Work at office
    Local area
    Relocation package
    Flexible hours

    Slope

    San Francisco, CA
    3 days ago
  • Together AI in San Francisco is seeking a Staff ML Systems Engineer to design and prototype algorithms, architectures, and scheduling for low-latency, high-throughput...  ...in production-grade inference engines, including kernel backends and ATLAS-style systems, while profiling... 

    Together

    San Francisco, CA
    4 days ago
  • $30 per hour

    AI Trainer - Advanced Japanese Fluency Prolific is building the biggest pool of quality human data in the world. Over 35,000 AI developers, researchers, and organizations use us to gather data from paid study participants with a wide variety of experiences, knowledge,... 
    Remote job
    Hourly pay
    Self employment
    Work from home
    Flexible hours

    Prolific

    San Francisco, CA
    3 days ago
  •  ...of AGI. Join us in shaping the future of AI and pushing the boundaries of what's possible...  ....We're seeking a GPU Performance Engineer to squeeze every last FLOP from our H100...  ...5-10x speedups. From writing custom CUDA kernels to eliminating cold start latency, you'll... 

    Genmo

    San Francisco, CA
    3 days ago
  • $170k - $205k

     ...intelligence. As the only vertically integrated AI infrastructure company built from the...  ..., hands-on Senior level performance engineer who thrives on deep technical challenges....  ...ideal candidate possesses a mastery of Linux kernel systems and low-level optimization, with... 
    Full time
    Temporary work

    Crusoe

    San Francisco, CA
    2 days ago
  • Armada is seeking a patient, articulate Part-Time Technical Trainer to deliver hands-on training and create demo content for our GPUaaS platform. This role combines technical enablement with customer-facing teaching, traveling to sites worldwide when required, and producing... 
    Part time
    Remote work
    Worldwide

    Armada

    San Francisco, CA
    1 day ago
  • Mercor, in collaboration with a leading AI lab, seeks experienced FP&A and treasury professionals to translate real planning and cash management work into structured training data for AI that reasons like finance teams. You will design scenarios such as budgets, rolling... 

    Obsidian

    San Francisco, CA
    1 day ago
  • $40 per hour

    Prolific is seeking AI Trainers with advanced SQL development skills to train and evaluate AI models. The role requires strong attention to detail, ability to work on complex tasks, and a reliable internet connection. Successful candidates will join Prolific as Domain Expert... 
    Remote job
    Flexible hours

    Prolific

    San Francisco, CA
    5 days ago
  •  ...Industrial, powered by Clean Harbors, is looking for a Data Center Sales Engineer to join their safety conscious team! The Sales Engineer has...  ...company.Please be advised that Artificial Intelligence (AI) tools may be used to support certain stages of the recruitment... 
    Full time
    For contractors
    Local area
    Remote work

    Clean Harbors

    San Francisco, CA
    5 days ago
  • $120k - $148k

    Joining ArupWe are seeking a Senior Electrical Engineer to join one of either our Los Angeles or San Francisco office with a focus on data center project delivery. This role involves leading electrical engineering design, collaborating with multidisciplinary teams, and... 
    Work at office
    Remote work
    Flexible hours

    Arup

    San Francisco, CA
    3 days ago
  • $195k - $225k

     ...abundance of energy and intelligence. As the only vertically integrated AI infrastructure company built from the ground up, we own and...  ...the Role: We are seeking a highly motivated Electrical Design Engineer to act as the primary technical bridge between internal... 
    Temporary work
    Work at office

    Crusoe

    San Francisco, CA
    4 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Trainium Kernel Engineer - NKI AI Trainer. Be the first to apply!