其他 金融业 50-200人 上海市宝山区
面议 上海市浦东新区 全职 本科及以上 5人 2025-08-31
2024-08-26 16:03:01
机器学习性能工程师将致力于结合量化行业的计算需求特征,探索不同硬件平台上的高性能计算技术,打造一流的高性能计算平台,同时优化机器学习基础设施、模型训练和推理,以改进我们的交易策略。
岗位职责
● 研究CPU/GPU等硬件平台上的高性能计算技术,参与高性能计算库、推理引擎的开发与优化。
● 参与机器学习平台的性能优化工作,包括计算资源调度策略设计,PB级高性能存储的存取加速,网络数据并行传输优化等。
● 负责模型训练优化,提升模型训练速度与平台利用效率,保持与业界先进技术同频。
● 构建模型的生命周期管理,提供模型仓库服务,保障交易场景模型的安全可靠加载。
● 探索前沿技术,持续迭代相关系统的性能。
任职要求
● 重点本科及以上学历,计算机、软件工程、电子工程、数学等理工科相关专业。
● 熟练掌握C/C++/Python中至少一门编程语言,有扎实的算法与数据结构基础,良好的代码风格和工作习惯。
● 熟练掌握计算机体系结构相关知识,熟悉现代处理器CPU/GPU的微架构。
● 熟练掌握Linux环境系统编程,具备多核并行编程及优化能力,熟练使用常见性能分析工具。
● 熟悉至少一种主流深度学习框架(TensorFlow/Pytorch/PaddlePaddle等),熟悉其底层架构和实现机制,了解其性能瓶颈分析与性能调优的方法。
● 有以下至少一项的背景知识与经验:GPU编程,并行计算,分布式存储,高性能网络,深度学习编译器。
加分项:
● 具备汇编层面的开发/优化能力,有CUDA开发经验。
● 在高性能计算和机器学习相关的开源项目中做出过相关贡献。
● 有大规模训练任务和推理服务编排和调度、镜像加速等工作经验优先。
● 有深度学习编译器开发经验优先,TVM/XLA等。
● 熟悉InfiniBand或RDMA。
职位类别:计算机软、硬件/互联网/IT
专业要求:不限
宽德投资是一家国内领先、业务全面的量化对冲基金。基于先进高效的研究和交易构架,以及完善的资产管理系统,宽德投资在国内期货、股票、期权等主流市场具有良好的盈利能力。 成立十年来,我们始终怀着打造世界顶尖华人量化对冲基金的梦想,低调扎实地工作,并热切期待着每位同路人。
请稍候...