高级/资深机器学习训练优化工程师
Senior / Staff ML Training Optimization Engineer
Waabi由人工智能先驱Raquel Urtasun创立，是物理AI领域的领导者。凭借世界级的团队，我们正在通过技术开启自动驾驶运输的新时代，这些技术正在为商用自动驾驶卡车和机器人出租车提供支持。Waabi与人工智能、汽车、物流和深度科技领域的世界领先企业合作并获得其支持。
在多伦多、旧金山、达拉斯和匹兹堡设有办公室，Waabi正在快速成长，寻找具有多样性、创新性和协作精神的候选人，他们希望以积极的方式影响世界。如需了解更多，请访问：www.waabi.ai
你将...
- 构建标准化的分布式训练框架用于研究和生产，推动我们的训练达到更高的稳定性和效率。
- 全面分析模型运行时和内存，定位性能瓶颈。
- 识别并评估可纳入Waabi训练和推理框架的新兴技术。例如，设计新的CUDA内核、量化感知训练和推理，以及编译/部署技术。
- 与研究人员和机器学习工程师合作，制定最佳资源使用实践。
- 创建和改进工具和仪表板，确保工作的广泛采用。
资格要求：
- 计算机科学、机器人学和/或其他相关技术领域的硕士/博士或学士学位，且至少有4年行业经验。
- 在多种编程语言中具备扎实的编码能力，包括Python、C++或Rust。
- 具有深度学习框架（如PyTorch或Jax）的经验。
- 熟练使用工具如PyTorch Profiler和NVIDIA Nsight对CPU和GPU代码进行分析。
- 思维开放、善于协作，愿意帮助他人。
- 对自动驾驶技术充满热情，喜欢解决难题并创造创新解决方案。
加分项：
- 具备识别何时需要自定义CUDA内核并实现它们的经验。
- 在单体仓库环境中使用Bazel，并将第三方包集成到开发环境中的经验。
- 使用基于Kubernetes的训练平台的经验。
该职位的美国年薪范围为：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...
- Build standardized distributed training frameworks for research and production, drive our training towards new levels of stability and efficiency.
- Comprehensively profile model runtime and memory to pinpoint performance bottlenecks.
- Identify and evaluate emerging technologies that can be adopted into Waabi’s training and inference frameworks. Examples include designing new CUDA kernels, quantization-aware training and inference, and compilation/deployment techniques.
- Work with researchers and ML engineers on best-practices for optimal resource usage.
- Create and improve tooling and dashboards to ensure broad adoption of your work.
Qualifications:
- MS/PhD or Bachelors degree with a minimum of 4 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 or Jax.
- Skilled in profiling CPU and GPU code using tools such as PyTorch Profiler and NVIDIA Nsight.
- 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 identifying when custom CUDA kernels are needed, and implementing them.
- Experience in Bazel in a monorepo environment, and integrating third party packages into dev environments.
- Experience with Kubernetes-based training platforms.
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.