#168 · 主分类： 深度学习框架
dfdx
autodiff
autodifferentiation
autograd
backpropagation
cuda
cuda-kernels
cuda-support
cuda-toolkit
cudnn
deep-learning
deep-neural-networks
gpu
gpu-acceleration
gpu-computing
machine-learning
neural-network
rust
rust-lang
tensor
用Rust进行深度学习，具有形状检查的张量和神经网络
项目最后更新：07/23/24
GitHub Stars
1.9K
Forks数量
108
贡献者数量
40
许可证
Other
收录理由
dfdx 是少数真正利用 Rust 类型系统而非简单模仿 PyTorch 的深度学习库。它的核心卖点在于编译期形状检查：如果模型里的张量维度不匹配，代码根本无法编译，这样在运行前就能发现错误。除此之外，它提供了扎实的张量库、自动微分引擎、常见的神经网络层（线性层、卷积、Transformer），以及 SGD、Adam、AdamW 等优化器，还可以启用 CUDA 进行 GPU 加速。对于想留在 Rust 生态里构建模型，或者需要编译期替代方案来移植研究代码的开发者来说，dfdx 是个不错的选择。不过要注意，项目仍处于预 alpha 阶段，计划发布破坏性更新，所以更适合用来学习和原型验证，而不是直接投入生产。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容，欢迎 联系我们推荐选题或投稿。