岗位职责
岗位职责:1、完善基于LibTorch的深度学习框架,优化网络层定义、模型训练与预测流程,提升框架的易用性与扩展性。2、完善现有的多种自定义训练参数(如epoch、优化器、损失函数等),设计灵活的API接口,降低用户使用门槛。3、完善现有的迁移学习功能,支持加载预训练模型、模型微调及跨任务训练。4、针对现有的CPU框架进行拓展,完善GPU加速版本,实现CPU/GPU的高效运行。5、编写清晰的技术文档,包括但不限于设计文档、API参考手册、用户使用指南等。任职要求:1、熟练掌握C++编程语言,熟悉多线程编程、内存管理等,具备扎实的工程实践能力;2、具备大型C++项目开发经验,熟悉CMake、Git等工具链。3、深入理解深度学习原理,熟悉常见网络结构(CNN、RNN、Transformer等)及训练流程;4、熟练掌握LibTorch或PyTorchPython端开发,有自定义网络层、优化器或损失函数的实现经验;5、有深度学习框架的二次开发经验者优先。行业要求:全部行业
工作地址
九龙坡区-九龙坡区 (重庆-九龙坡区北京大学重庆大数据研究院) 查看地图
