Show HN: Cant, rust nn lib for learning
10 months ago
- #rust
- #pytorch-like
- #machine-learning
- 一个用Rust编写的机器学习库,提供类似PyTorch的功能。
- 功能包括自动微分、张量运算和GPU加速。
- 支持使用ndarray进行CPU运算,并为macOS提供Metal GPU加速。
- 包含类似PyTorch的API以便于使用。
- 项目设置包括测试、构建、代码格式化和覆盖率报告的指令。
- 核心组件:Tensor结构体、计算图管理器、运算定义和形状处理。
- 后端支持包括CPU(ndarray)和Metal GPU(macOS)。
- 测试涵盖单元测试、模型测试和运算验证。
- 采用MIT许可证授权。