Machine Learning Engineer (AI infra)
base地设定在上海,全职和实习皆可,欢迎全球各地优秀的华人加入。 【关于衍复】 上海衍复投资管理有限公司成立于2019年,是一家用量化方法从事投资管理的科技公司。 公司策略团队成员的背景丰富多元:有曾在海外头部对冲基金深耕多年的行家里手、有在美国大学任教后加入业界的学术型专家以及国内外顶级学府毕业后在衍复成长起来的中坚力量;工程团队核心成员均来自清北交复等顶级院校,大部分有一线互联网公司的工作经历,团队具有丰富的技术经验和良好的技术氛围。 公司致力于通过10-20年的时间,把衍复打造为投资人广泛认可的头部资管品牌。 衍复鼓励充分交流合作,我们相信自由开放的文化是优秀的人才发挥创造力的土壤。我们希望每位员工都可以在友善的合作氛围中充分实现自己的职业发展潜力。 【工作职责】 1、负责机器学习/深度学习模型的研发,优化和落地,以帮助提升交易信号的表现; 2、研究前沿算法及优化技术,推动技术迭代与业务创新。 【任职资格】 1、本科及以上学历,计算机相关专业,国内外知名高校; 2、扎实的算法和数理基础,熟悉常用机器学习/深度学习算法(XGBoost/LSTM/Transformer等); 3、熟练使用Python/C++,掌握PyTorch/TensorFlow等框架; 4、具备优秀的业务理解能力和独立解决问题能力,良好的团队合作意识和沟通能力。 【加分项】 1、熟悉CUDA,了解主流的并行编程以及性能优化技术; 2、有模型实际工程优化经验(如训练或推理加速); 3、熟悉DeepSpeed, Megatron等并行训练框架; 4、熟悉Triton, cutlass,能根据业务需要写出高效算子; 5、熟悉多模态学习、大规模预训练、模态对齐等相关技术。