Your browser does not support javascript! Please enable it, otherwise web will not work for you.

GPU Kernel Engineer

Home > C/C++ programming jobs

GPU Kernel Engineer in USA new

  • Sciforium
  • Full time
  • Email
  • San Francisco, CA

Responsibilities

  • Design, implement, and optimize custom GPU kernels using C++, PTX, CUDA, ROCm, Triton, and/or JAX Pallas.
  • Profile end-to-end ML operation performance for large-scale LLM training and inference.
  • Integrate low-level GPU kernels into PyTorch, JAX, and custom internal runtimes.
  • Develop performance models, identify bottlenecks, and deliver kernel-level performance improvements.
  • Collaborate with ML researchers, distributed systems engineers, model-serving teams, and NVIDIA/AMD hardware vendors.
  • Contribute to tooling, documentation, benchmarking suites, and testing frameworks for correctness and reproducibility.

Requirements

  • At least 5 years of industry or research experience in GPU kernel development or high-performance computing.
  • Bachelor’s, Master’s, or PhD in Computer Science, Computer Engineering, Electrical Engineering, Applied Mathematics, or a related field.
  • Strong programming skills in C++ and Python, with familiarity with ML frameworks.
  • Deep expertise in CUDA/ROCm, GPU memory models, and performance optimization.
  • Hands-on experience with Triton and/or JAX Pallas for custom kernel development.
  • Strong understanding of PTX, GPU ASM, and low-level GPU execution.
  • Extensive experience writing and optimizing custom GPU kernels in C++ and PTX.
  • Experience integrating low-level kernels into PyTorch, JAX, or similar frameworks.
  • Experience with large-scale LLM training or inference.
  • Preferred experience with AMD GPUs, ROCm optimization, JAX FFI, custom ML operators, vLLM, TensorRT, TPUs, XLA, accelerator programming, compilers, or open-source ML systems.

Benefits

  • Medical, dental, and vision insurance.
  • 401(k) plan.
  • Daily lunch, snacks, and beverages.
  • Flexible time off.
  • Competitive salary and equity.

Salary: $190k - $250k/yr

Sciforium

Sciforium is a dynamic marketplace that connects science fiction enthusiasts with a curated selection of books, collectibles, and fan experiences. We empower creators and fans alike by offering a platform that celebrates imagination, community, and the limitless possibilities of speculative story...

Similar positions

Senior Software Engineer, Simulation

  • ATOMS
  • Full time
  • USA
  • 09/13/2026
  • Salary: $195k - $246k/yr
  • San Francisco, CA

Firmware Engineer II

  • Varda Space Industries
  • Full time
  • USA
  • 09/13/2026
  • Salary: $144k - $184k/yr
  • El Segundo, CA

Senior Software Engineer, Simulation

  • CloudKitchens
  • Full time
  • USA
  • 09/13/2026
  • Salary: $195k - $246k/yr
  • San Francisco, CA

Software Engineer, Battlespace Awareness

  • Anduril Industries
  • Full time
  • USA
  • 09/13/2026
  • Salary: $191k - $253k/yr
  • Waltham, MA

Software Engineer, Battlespace Awareness

  • Anduril Industries
  • Full time
  • USA
  • 09/13/2026
  • Salary: $165k - $218k/yr
  • Broomfield, CO

Listings

by


Job Type

Language

Country

CodingJobBoard.com © 2026