MLIR后端开发工程师,高级员工

16个月前全职
d-Matrix

d-Matrix

location 悉尼
unsaved
d-Matrix基于我们的数字内存计算(DIMC)引擎从根本上改变了内存计算集成的物理特性。AI计算的“圣杯”一直是突破内存壁垒以最小化数据移动。我们通过首创的DIMC引擎实现了这一目标。d-Matrix已经获得了超过1.54亿美元的资金,其中1.1亿美元来自我们的B轮融资,d-Matrix准备利用我们的芯片和内存计算方法推进大型语言模型的生成推理加速。我们计划在2024年初推出我们的第一款商业产品。我们准备满足这些大型语言模型的能源和性能需求。公司在硅谷、悉尼和班加罗尔拥有70多名员工。 地点: 在我们澳大利亚悉尼总部上班。 职位:MLIR后端开发工程师,高级 你将做什么: d-Matrix编译器团队正在寻找杰出的候选人来帮助开发编译器后端 - 具体来说是在空间架构中分配硬件资源以执行低级指令的问题。成功的候选人将有动力,能够解决算法编译器问题,并有兴趣学习底层硬件和软件架构的复杂细节。成功的候选人将加入一支经验丰富的编译器开发团队,该团队将指导候选人快速掌握编译器基础设施,以解决将低级指令映射到硬件资源的重要问题。我们在以下领域有特定的机会: •模型分区(流水线、张量、模型和数据并行)、平铺、资源分配、内存管理、调度和优化(用于延迟、带宽和吞吐量)。 你需要具备什么: 最低要求: •相关领域的学士学位。 •能够使用现代C++编写生产质量的代码。 •具有现代编译器基础设施的经验,例如:LLVM,MLIR。 •具有机器学习框架和接口的经验,例如:ONNX,TensorFlow和PyTorch。 •具有生产编译器开发经验。 首选: •从高级概念设计到实际实现的算法设计能力。 •具有与Torch-MLIR、ONNX-MLIR、Caffe、TVM等相关的开源ML项目经验。 •对快节奏和充满活力的初创企业文化充满热情。 平等就业政策 d-Matrix自豪地成为一个平等机会的工作场所和积极行动的雇主。我们致力于营造一个包容的环境,让每个人都感到受欢迎和有能力做出最好的工作。我们为团队招聘最优秀的人才,不论种族、宗教、肤色、年龄、残疾、性别认同、性取向、血统、遗传信息、婚姻状况、国籍、政治隶属或退伍军人身份。我们的重点是聘用具有谦逊的专业知识、友善、敬业和愿意共同面对挑战并共同学习的团队成员。 d-Matrix不接受外部机构的简历或候选人提交。我们感谢招聘公司的兴趣和努力,但我们恳请有意向加入d-Matrix的个人直接通过我们的官方渠道申请。这种方法可以使我们简化招聘流程,并保持对所有申请者的一致和公正评估。感谢您的理解和合作。