Hasty Briefsbeta

双语

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许可证授权。