LHaiC / Programming_in_AILinks

​​北京大学《人工智能中的编程》2024年课程作业代码:基于CUDA和PyBind11的简易CNN训练框架​​。使用C++/CUDA编写高性能计算内核,通过PyBind11封装为Python库,实现了基础的类PyTorch训练功能。当前版本较为精简,仅供参考学习。
13Updated last month

Alternatives and similar repositories for Programming_in_AI

Users that are interested in Programming_in_AI are comparing it to the libraries listed below

Sorting: