职位描述
高效且稳定的AI基础架构和开发工具链,是领先的策略模型能够诞生和不断优化的重要基础。我们致力于打造一流的高性能计算平台,不断优化AI基础设施、模型训练和推理服务,探索金融市场的未来。
你将会
- 参与量化交易模型训练框架的开发及性能优化工作;
- 优化CPU/GPU的推理延迟,根据业务编排算子,组织计算流水线,优化算子的执行效率,实现等效可替换算子等工作;
- 提高数据加载速度、缓存利用率,优化分布式训练流程,提升训练中的GPU利用率;
- 设计和实现各类机器学习训练工具,简化从回测到实盘的部署流程,方便研究员管理模型、数据、算法等模块,加速模型的开发和测试。
职位要求
我们希望你
- 国内外知名院校本科及以上学历,计算机、通信、软件工程等计算机相关专业;
- 熟悉C++、CUDA、Python等语言,扎实代码功底和实战能力;
- 熟练掌握Tensorflow/Pytorch之一,熟悉分布式训练架构与调优;
- 熟悉高性能 RDMA 通信等相关知识。
加分项
- 在IOI,NOI,ICPC,CCPC,TopCoder等竞赛中获奖;
- 熟悉分布式计算/异构计算,有低延迟/高性能计算相关经验;
- 有模型生命周期管理、模型上下线流水线设计经验;
- 有参与大型开源项目的经验。