Developer Technology Engineer - AI
Core
Build and optimize core parallel algorithms, data structures, and high-performance operators for GPUs to support large language model (LLM) training and inference, while improving distributed communication for broad workloads.
Role type
Senior IC GPU software engineer (LLM optimization)
Builds
Optimized GPU kernels, communication libraries, and frameworks for LLMs
Domain
Artificial Intelligence / High-Performance Computing / GPU Architecture
Deliverable
production ML models
Required skills
C/C++/Python/Fortran, parallel programming, GPU programming, algorithm design, distributed systems, compiler optimization, linear algebra, numerical methods
Preferred skills
LLM framework optimization, distributed training/inference, interconnect topology design, open-source contribution
Technologies
Megatron, TRTLLM, SGLang, vLLM, cuDNN, cuBLAS, CUTLASS, DeepGEMM, FlashMLA, FlashAttention, Flashinfer, NCCL, NVSHMEM, NVLINK, InfiniBand, RoCE
Responsibilities
Build and optimize core parallel algorithms and data structures for GPUs; Engage in deep optimization of high-performance operators (GPU kernel, instruction-level, compiler); Improve communication for distributed LLM workloads; Collaborate with architecture and research teams to influence next-generation platforms.
Seniority
Senior, hands-on IC
