机器学习 onboard 优化工程师
ML Onboard Optimization Engineer
Waabi由人工智能先知Raquel Urtasun创立，是物理AI领域的领导者。我们拥有世界级的团队，正在通过技术开启自动驾驶运输的新时代，这些技术正在为商用自动驾驶卡车和机器人出租车提供动力。Waabi得到了人工智能、汽车、物流和深度科技领域世界领先企业的支持与合作。
Waabi在多伦多、旧金山、达拉斯和匹兹堡设有办公室，发展迅速，正在寻找具有多样性、创新性和协作精神的候选人，他们希望以积极的方式影响世界。如需了解更多信息，请访问：www.waabi.ai
你将...
- 与自动驾驶和算法工程师紧密合作，采用以AI为核心的方法，扩展安全的自动驾驶系统。
- 将模型部署流程扩展到新的GPU和嵌入式系统，用于我们车载计算系统的下一代。
- 使用TensorRT和modelopt等框架优化运行在卡车上的模型。
- 为推理创建并基准测试新的CUDA内核。
- 对模型运行时间和内存进行全面分析，找出性能瓶颈。
资格要求：
- 计算机科学、机器人学和/或其他相关技术领域的硕士/博士或学士学位，且至少有6年行业经验。
- 在多种编程语言中具备扎实的编码能力，包括Python、C++或Rust。
- 具有PyTorch等深度学习框架的经验。
- 精通使用PyTorch Profiler和NVIDIA Nsight等工具对CPU和GPU代码进行分析。
- 具有Nvidia嵌入式平台（如Nvidia Jetson或Thor）的经验。
- 思维开放、善于协作，愿意帮助他人。
- 对自动驾驶技术充满热情，喜欢解决难题并创造创新解决方案。
加分项/有帮助的：
- 具有模型编译和导出的经验，与TensorRT等底层概念进行交互。
- 具有识别是否需要自定义CUDA内核并实现它们的经验。
- 具有Bazel构建系统经验，并能将第三方包集成到开发环境中。
该职位在美国的年薪范围为：141,000美元至249,000美元，另加具有竞争力的福利和待遇。Waabi的年薪范围根据公司薪酬政策，结合多个因素确定。该薪资范围反映了该职位在美国所有地区的新人薪资的最低和最高目标。注意：公司会为该职位的员工提供额外的补偿。
查看英文原文
Waabi, founded by AI visionary Raquel Urtasun, is the leader in Physical AI. With a world-class team, we're unlocking the next era of autonomous transportation with technology that's powering commercial autonomous trucks and robotaxis. Waabi is backed by and partners with world leaders in AI, automotive, logistics, and deep tech.
With offices in Toronto, San Francisco, Dallas, and Pittsburgh, Waabi is growing quickly and looking for diverse, innovative and collaborative candidates who want to impact the world in a positive way. To learn more visit: www.waabi.ai
You will...
- Collaborate closely with autonomy and algorithm engineers to scale safe self-driving systems using an AI-first approach.
- Expand the model deployment pipeline to new GPUs and embedded systems for the next generation of our onboard compute system.
- Use frameworks such as TensorRT and modelopt to optimize the models running on the truck.
- Create and benchmark new CUDA kernels for inference.
- Comprehensively profile model runtime and memory to pinpoint performance bottlenecks.
Qualifications:
- MS/PhD or Bachelors degree with a minimum of 6 years of industry experience in Computer Science, Robotics and/or similar technical field(s) of study.
- Solid coding proficiency in a variety of coding languages including Python, C++ or Rust.
- Experience in deep learning frameworks such as PyTorch.
- Skilled in profiling CPU and GPU code using tools such as PyTorch Profiler and NVIDIA Nsight.
- Experience with Nvidia embedded platforms such as Nvidia Jetson or Thor.
- Open-minded and collaborative team player with willingness to help others.
- Passionate about self-driving technologies, solving hard problems, and creating innovative solutions.
Bonus/nice to have:
- Experience in model compilation and exporting, interaction with lower level concepts like TensorRT.
- Experience in identifying when custom CUDA kernels are needed, and implementing them.
- Experience in Bazel build systems, and integrating third party packages into dev environments.
The US yearly salary range for this role is: $141,000 - $249,000 in addition to competitive perks & benefits. Waabi’s yearly salary ranges are determined based on several factors in accordance with the Company’s compensation practices. The salary base range is reflective of the minimum and maximum target for new hire salaries for the position across all US locations. Note: The Company provides additional compensation for employees in this role, including equity incentive awards and an annual performance bonus.
Perks/Benefits:
Waabi provides a competitive benefits package that includes:
Competitive compensation and equity awards.
Health and Wellness benefits encompassing Medical, Dental and Vision coverage.
Unlimited Vacation.
Flexible hours and Work from Home support.
Daily drinks, snacks and catered meals (when in office).
Regularly scheduled team building activities and social events both on-site, off-site & virtually.
World-class facility that includes a gym, games room (ping pong table, video game consoles, board games, etc), multiple collaborative working spaces and a gorgeous patio!(when in office)
As we grow, this list continues to evolve!
Waabi is an equal opportunity employer that celebrates diversity and is committed to creating a supportive, inclusive, and accessible environment for all employees. If reasonable accommodation is needed to participate in the job application or interview process please let our recruiting team know.