高级/资深机器学习优化工程师

蒙特利尔 6天前全职 网络
面议
Waabi由人工智能先驱和愿景家Raquel Urtasun创立,是一家致力于构建下一代自动驾驶技术的人工智能公司。凭借一支世界级的团队和创新的方法,Waabi释放了人工智能在现实世界中“安全驾驶”的力量,将自动驾驶的承诺比以往任何时候都更接近商业化。Waabi得到了技术、物流和加拿大创新生态系统中顶级投资者的支持。Waabi在多伦多、旧金山、达拉斯和匹兹堡设有办事处,正在快速发展,寻找多元化、创新和协作的候选人,希望以积极的方式影响世界。欲了解更多信息,请访问:您的职责包括:- 与自动化和算法工程师密切合作,采用AI优先的方法扩展安全自动驾驶系统。- 为研究和生产构建标准化的分布式训练框架,将我们的训练推向新的稳定性和效率水平。- 全面分析模型运行时和内存,以查明性能瓶颈。- 识别和评估可纳入Waabi训练和推理框架的新兴技术,包括但不限于用于训练的高效CUDA内核、量化、模型导出和推理编译等。资格要求:- 拥有计算机科学、机器人技术和/或类似技术领域的MS/PhD或学士学位,并具有至少6年的行业经验。- 在包括Python、C++或Rust在内的多种编程语言中具有扎实的编码能力。- 具有PyTorch等深度学习框架的经验。- 在开发管道的不同阶段具有经验:数据处理、分布式训练和模型部署。- 熟练使用PyTorch Profiler和NVIDIA Nsight等工具分析CPU和GPU代码。- 思维开放且具有协作精神的团队成员,愿意帮助他人。- 对自动驾驶技术充满热情,致力于解决难题并创造创新解决方案。加分项/优先考虑:- 具有模型编译和导出、与TensorRT等低级概念交互的经验。- 具有识别何时需要自定义CUDA内核并实现它们的经验。- 具有Bazel构建系统的经验,并将第三方软件包集成到开发环境中。该职位在美国的年薪范围为:$141,000 - $249,000,此外还有竞争力的福利待遇。Waabi的年薪范围是根据公司的薪酬实践根据多个因素确定的。薪资基准范围反映了该职位在美国所有地点的新员工薪资的最低和最高目标。注意:公司为该职位的员工提供额外的补偿,包括股权激励奖励和年度绩效奖金。福利/待遇:- 具有竞争力的薪酬和股权奖励- 包括医疗、牙科和视力保险的健康和保健福利- 无限假期- 灵活的工作时间和在家办公支持- 每日饮料、小吃和餐饮(在办公室时)- 定期安排的团队建设活动和社交活动,包括现场、非现场和虚拟活动- 世界级设施,包括健身房、游戏室(乒乓球桌、电子游戏机、棋盘游戏等)、多个协作工作空间和一个美丽的露台!(在办公室时)- 随着我们的成长,这个清单将不断发展!Waabi是一个平等机会的雇主,庆祝多样性,致力于为所有员工创造一个支持性、包容性和无障碍的环境。如果在求职申请或面试过程中需要合理的便利,请告知我们的招聘团队。我们可能会使用人工智能(AI)工具来支持招聘过程的某些部分,例如审查申请、分析简历或评估回应。这些工具协助我们的招聘团队,但不会取代人类的判断。最终的招聘决定由人类做出。如果您想了解有关您的数据处理的更多信息,请联系我们。