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

15个月前全职
d-Matrix

d-Matrix

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