蒸馏负责人
Distillation Lead
Waabi由人工智能愿景家Raquel Urtasun创立,是物理AI领域的领导者。凭借世界级的团队,我们正在通过技术开启自动驾驶运输的新时代,这些技术正在为商用自动驾驶卡车和机器人出租车提供支持。Waabi得到了人工智能、汽车、物流和深度科技领域世界领先企业的投资并与其合作。
Waabi在多伦多、旧金山、达拉斯和匹兹堡设有办公室,正在快速扩张,寻找具有多样性、创新性和协作精神的候选人,希望以积极的方式影响世界。如需了解更多信息,请访问:www.waabi.ai
Waabi的物理AI平台由最先进的机器学习模型驱动,这些模型必须在各种使用场景中高效部署,包括车载推理和大规模模拟。作为蒸馏负责人,你将负责Waabi AI架构中的蒸馏策略和执行,确保我们的最先进模型在所有部署环境中都能高效运行。你将与机器学习平台、基础设施、车载自主和模拟团队紧密合作,交付符合实时车载系统和高吞吐量模拟流水线性能要求的压缩模型。
你将……
定义并推动Waabi AI架构中模型蒸馏和压缩的技术战略——涵盖感知、世界模型和规划——同时关注车载部署和模拟使用场景。
设计、实现并扩展最先进的蒸馏和效率流程,可能包括:
生成模型的蒸馏(扩散、自回归、流匹配、视频模型)
量化感知训练(QAT)和训练后量化(PTQ)
知识蒸馏(特征级、响应式和关系式)
结构化和非结构化剪枝与稀疏化
低秩分解和高效架构设计
推测解码和其他推理时效率技术
与机器学习平台、基础设施、车载、自主和模拟团队密切合作,将压缩模型集成到生产流程中,并在不同部署环境中满足延迟、内存和吞吐量目标。
定义严格的基准测试和评估框架,以分析模型和硬件目标之间的效率与质量权衡。
指导和引导从事蒸馏和模型效率工作的研究人员和工程师,设定高标准的技术要求,并培养严谨实验的文化。
查看英文原文
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
Waabi’s Physical AI platform is powered by state of the art ML models which must be deployed efficiently across diverse use-cases, from onboard vehicle inference to large-scale simulation. As the Distillation Lead, you will own the strategy and execution for distillation across Waabi's AI stack, ensuring our most capable models run efficiently in every deployment context. You will partner closely with ML Platform, Infrastructure, Onboard Autonomy, and Simulation teams to deliver compressed models that meet the performance requirements of both real-time onboard systems and high-throughput simulation pipelines.
You will…
Define and drive the technical strategy for model distillation and compression across Waabi's AI stack — spanning perception, world models, and planning — with an eye toward both onboard deployment and simulation use-cases.
Design, implement, and scale state-of-the-art distillation and efficiency pipelines, which may include:
Distillation for generative models (diffusion, autoregressive, flow-matching, video models)
Quantization-aware training (QAT) and post-training quantization (PTQ)
Knowledge distillation (feature-level, response-based, and relation-based)
Structured and unstructured pruning and sparsification
Low-rank factorization and efficient architecture design
Speculative decoding and other inference-time efficiency techniques
Collaborate closely with ML Platform, Infrastructure, Onboard, Autonomy, and Simulation teams to integrate compressed models into production pipelines and meet latency, memory, and throughput targets across deployment contexts.
Define rigorous benchmarks and evaluation frameworks to characterize efficiency vs. quality trade-offs across models and hardware targets.
Mentor and guide researchers and engineers working in the distillation and model efficiency space, setting a high technical bar and fostering a culture of rigorous experimentation.
Champion best practices for model compression across the organization; disseminate knowledge through internal design reviews, documentation, and technical talks.
Stay at the cutting edge of model efficiency research; contribute to the broader scientific community through publications and open-source contributions.
Qualifications:
Deep distillation expertise: You have extensive hands-on experience designing and implementing distillation, quantization, pruning, and model compression techniques for large-scale neural networks, with demonstrated impact in production settings.
Strong research and engineering foundation: A Bachelor's or Master's degree in Machine Learning, Computer Vision, Robotics, or a related field, or equivalent industry experience; relevant hands-on experience in model distillation and efficiency is what matters most. Expert Python and PyTorch (or JAX) skills with experience in large-scale distributed training.
Technical leadership: You have a proven track record of setting technical direction and driving projects from conception to production. You inspire and elevate those around you through deep technical expertise and mentorship.
Cross-functional collaboration: You have experience working closely with infrastructure, platform, and autonomy teams to deploy compressed models under real engineering constraints.
Clear communicator: You can communicate complex technical trade-offs clearly to diverse audiences and drive alignment across research and engineering teams.
Bonus:
Experience with hardware-aware optimization (TensorRT, ONNX, custom CUDA kernels, hardware-specific quantization).
Publications at top-tier ML/CV venues (NeurIPS, ICML, CVPR, ICLR, ECCV) in model compression, efficient deep learning, or related areas.
Experience distilling large generative models (diffusion models, LLMs, VLMs, or video models).
Background in autonomous vehicles or robotics.
The US yearly salary range for this role is: $195,000 - $286,000 USD in addition to competitive perks & benefits. Waabi (US) Inc.’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:
Competitive compensation and equity awards.
Health and Wellness benefits encompassing Medical, Dental and Vision coverage (for full-time employees only).
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.
As we grow, this list continues to evolve!
Waabi is a technology start-up building technologies to transform the way the world moves. Join our talented team to be a part of the future and to make an impact!
Waabi is an equal opportunity employer. We celebrate diversity and are committed to creating a supportive, inclusive, and accessible workplace for all our employees. We seek applicants of all backgrounds and identities, across race, color, ethnicity, national origin or ancestry, age, citizenship, religion, sex, sexual orientation, gender identity or expression, military or veteran status, marital status, pregnancy or parental status, caregiver status, disability, or any other characteristic protected by law. We make workplace accommodations for qualified individuals with disabilities as required by applicable law. If reasonable accommodation is needed to participate in the job application or interview process please let our recruiting team know.