#neural-network #opencl #cuda #machine-learning #deep-learning

RayBNN_Neural

使用CUDA、OpenCL和oneAPI通过GPU、CPU和FPGA在Rust中实现稀疏权重的神经网络

6个版本 (2个稳定版)

2.0.3 2024年6月19日
2.0.1 2024年6月5日
0.1.3 2024年6月5日
0.1.2 2024年5月19日

#483 in 机器学习

Download history 414/week @ 2024-05-15 56/week @ 2024-05-22 9/week @ 2024-05-29 298/week @ 2024-06-05 4/week @ 2024-06-12 144/week @ 2024-06-19 13/week @ 2024-07-03 2/week @ 2024-07-10

每月下载量443次

GPL-3.0-only

44KB
677

RayBNN_Neural

使用CUDA、OpenCL和oneAPI通过GPU、CPU和FPGA在Rust中实现稀疏权重的神经网络

安装Arrayfire

https://arrayfire.com/binaries/安装Arrayfire 3.9.0的二进制文件

或从源代码构建https://github.com/arrayfire/arrayfire/wiki/Getting-ArrayFire

添加到Cargo.toml

arrayfire = { version = "3.8.1", package = "arrayfire_fork" }
rayon = "1.10.0"
num = "0.4.3"
num-traits = "0.2.19"
half = { version = "2.4.1" , features = ["num-traits"] }
RayBNN_DataLoader = "2.0.3"
RayBNN_Graph = "2.0.3"
RayBNN_Sparse = "2.0.2"
RayBNN_Optimizer = "2.0.1"
RayBNN_Neural = "2.0.3"

示例列表

依赖项

~8.5MB
~162K SLoC