任务详情
详细介绍： 优化神经网络算法的实现，原实现使用pytorch GPU，现在希望能够用CUDA进行计算优化（例如底层代码用C++实现），开发一个类似于catboost https://catboost.ai/ 的python库。算法仓库详见 https://github.com/qile2000/LAMDA-TALENT， 方法为ModernNCA, paper: https://arxiv.org/abs/2407.03257。算法目前已有学术paper和代码支撑，仅需优化其实现效率。验收方式：通过算法实现的性能比较验收。价格可议。
