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

16个月前全职
d-Matrix

d-Matrix

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