Senior Software Architect Jobs

Senior Software Architect (GPU Kernels):

Seeking a Sr. Software Architect (GPU Kernels) to join a leading Artificial General Intelligence (AGI) startup who has just raised their Series A and is looking to scale their founding team. The team is in the process of building aligned and more complete AI to accelerate humanity’s progress on the world’s most important problems.

As a Sr. Software Architect (GPU Kernels), you'll be responsible for ensuring lightning-fast training and inference with giant neural networks. You'll implement high-throughput, low-latency GPU kernels in CUDA or Triton, optimize kernel fusions, and maximize multi-GPU communication through efficient overlapping of communication and computation.

Your work will help us build and safely deploy aligned AGI in products that truly matter. Are you ready to join us on this exciting journey?

What we can offer a Sr. Software Architect (GPU Kernels):

• Remote or hybrid work in NYC/ SF
• Highly competitive compensation (90th percentile)
• Strong equity + benefits package

Keywords: large language models, LLMs, GPU, GPU kernels, software architecture, machine learning, software engineering, startup, mixed-precision, training, inference, low-latency, CUDA, triton, kernel fusion