chelsea0x3b/dfdx

Deep learning in Rust, with shape checked tensors and neural networks

语言：Rust · GitHub 星标：1,942 · Fork：109

话题：autodiff · autodifferentiation · autograd · backpropagation · cuda · cuda-kernels · cuda-support · cuda-toolkit

在 GitHub 上查看原仓库