Jay Shah

Jay Shah 是 Colfax International 的加速计算研究科学家,Colfax International 是硅谷领先的定制计算解决方案提供商。他负责领导 Colfax 针对最新的 NVIDIA GPU 架构优化 ML/ AI 应用程序的工作。在加入 Colfax 之前,他在 2017 年获得了麻省理工学院数学博士学位,并担任过多个学术职务,从事拓扑和更高范畴理论的研究并发表论文。

Posts by Jay Shah

开发与优化

CUTLASS 3.x:用于 GEMM 内核设计的正交、可重用和组合抽象

GPU 上的 GEMM 优化是一个模块化问题。高性能实现需要指定超参数,例如图块形状、数学和复制指令以及线程束专用方案。 3 MIN READ