您的工作
DarkVision 正在寻找一名机器学习工程师加入我们的成像与人工智能团队。您将构建和维护推理和训练管道,以支持我们的机器学习工作负载。您将帮助将研究代码转化为稳健、可扩展的软件管道。
DarkVision 的超声成像系统收集了数量级为 TB 的庞大数据集,检测跨越数百公里的工业资产中的亚毫米缺陷。对这些 TB 级数据进行可靠和自动化处理至关重要。作为机器学习工程师,您将专注于软件基础设施的设计、实施和维护,使我们的模型能够可靠地训练并高效地推理。
该职位位于我们位于不列颠哥伦比亚省北温哥华的总部,员工可以享受各种设施,包括设备齐全的健身房、壁球场、蒸汽室、攀岩墙等!
我们的团队
在成像与人工智能团队工作,您将加入一个多学科的科学家和工程师团队。该团队负责早期阶段的构思、研究、实验和开发。您将与其他技术成员密切合作,确保我们的模型无缝集成到我们的产品中。
您的工作内容
• 推理管道工程:
开发和维护负责通过我们的模型处理数据的代码。您将确保推理任务的高效执行,重点关注计算资源的最佳使用。
• 训练基础设施和自动再训练:
设计和实施用于模型训练和再训练的自动化管道。您将构建允许重复、可扩展训练循环的系统。
• 生命周期管理:
建立和维护模型和数据集版本控制的最佳实践。您将实施跟踪模型谱系的工具,将特定模型版本与用于创建它们的确切数据和超参数连接起来。
• 数据集成:
编写与内部数据摄取系统接口所需的逻辑。您将处理数据的高效加载、预处理和移动,以确保管道正确供给。
您的基本资格
• 计算机科学、工程或相关领域的学士学位。
• 2 年以上软件工程或机器学习工程经验。
• 专业(生产代码)级别的 Python 熟练程度,专注于编写干净、模块化和经过测试的代码。
• 能够将实验代码转化为生产。
• 具备深度学习框架的经验,特别是 PyTorch。
• 理解高性能计算、并行处理和分布式系统。
加分条件
• 具有工作流编排工具的经验(例如,Prefect、Airflow 或 Dagster)。
• 熟悉分布式计算框架(例如,Ray、Dask、Monarch)。
• 具有用于实验跟踪和工件管理的 MLOps 工具经验(例如,Weights & Biases、DVC、MLFlow)。
• 具有模型优化和加速技术的经验,包括 TensorRT、ONNX 和 CUDA。熟悉量化和混合精度训练以及延迟和吞吐量的分析和调试。
• 了解 PyTorch Lightning、Hugging Face Accelerate 或类似的 ML 框架。
• 具备 SQL 和数据库的工作理解。
• 良好的沟通能力,能够向不同的技术团队清晰表达工程限制。
一般薪资范围
对于该职位,我们预计每年支付 $100,000 到 $150,000。该职位有资格获得可变薪酬,以货币奖金或其他形式发放。
在 Koch 公司,我们是企业家。这意味着我们公开挑战现状,寻找创造价值的新方法,并因个人贡献而获得奖励。为某个职位提供的任何薪酬范围都是根据可用的市场数据估算得出的。实际金额可能高于或低于所提供的范围,具体取决于每位候选人的知识、技能、能力和地理位置。如果您有任何疑问,请与您的招聘人员讨论我们薪酬理念的灵活性和细节。