2个版本
0.1.1 | 2023年1月17日 |
---|---|
0.1.0 | 2023年1月17日 |
#512 in 机器学习
7KB
171 行
rsgrad
进行中 - 尚未完成
micrograd但用Rust编写
遵循Andrej Karpathy的教程编写。
待办事项
- CUDA支持
- 实现算法
- 使用(half)https://github.com/starkat99/half-rs crate支持f16
为什么?
- 在Rust中重写很有趣
- 可能比Python快
- 对于项目,您可以拥有编译后的二进制文件而不是Python代码(例如:想象一个稳定的扩散作为一个二进制文件,而不是需要安装所有库)
- Wasm支持?
依赖关系
~465KB