CUDA内核工程师

Pragmatike代表一家快速成长的AI初创公司，由MIT CSAIL研究人员创立。我们正在寻找一名CUDA内核工程师，具有从零开始开发和优化NVIDIA CUDA内核的经验。你将参与驱动大规模、高吞吐量AI系统的核心GPU性能层，为福布斯500强客户提供解决方案。该职位适合深入理解NVIDIA GPU架构、内存层次结构、线程级执行及性能分析流程的人才，而非来自通用硬件、FPGA或非NVIDIA计算背景的候选人。

设计、实现和优化定制的CUDA内核，专注于提升占用率、内存吞吐量和线程效率；使用Nsight Compute、Nsight Systems、nvprof等工具分析GPU工作负载；排查性能瓶颈，包括线程发散、未合并内存访问、寄存器压力及PCIe传输开销；改进GPU内存管道（全局、共享、L2、纹理内存）并确保内存合并；与AI系统、模型加速和后端分布式系统团队紧密协作；参与GPU架构决策、内核库及性能工程最佳实践制定。

具有竞争力的薪资与股权期权、入职奖金、健康、牙科和视力保险、401k计划

Pragmatike是一家快速成长的AI初创公司，由MIT CSAIL研究人员创立，被GTM Capital评为年度前十名生成式AI公司。公司致力于推动AI系统突破，其校友已带领高价值收购（MosaicML Databricks、Run:AI Nvidia、W&B CoreWeave）。