Skip to main content

Rust写的深度学习框架该库旨在成为一个完整的深度学习框架,具有用 Rust 编写的极大灵活性

  1. Rust写的深度学习框架

    该库旨在成为一个完整的深度学习框架,具有用 Rust 编写的极大灵活性。 目标是满足研究人员和从业者的需求,使其更容易实验、训练和部署你的模型。

    特征:

    1.灵活直观的自定义神经网络模块
    2.无状态和线程安全正向传递
    3.快速培训,全面支持,以及metricloggingcheckpointing
    4.Burn-Tensor:支持自动比较、CPU 和 GPU 的张量库
    刻录数据集:具有多个实用程序和源的数据集库

    burn | #框架