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

RayBNN_Neural

Neural Networks with Sparse Weights in Rust using GPUs, CPUs, and FPGAs via CUDA, OpenCL, and oneAPI

5 releases (1 stable)

new 2.0.1 Jun 5, 2024
0.1.3 Jun 5, 2024
0.1.2 May 19, 2024
0.1.1 May 18, 2024
0.1.0 May 17, 2024

#357 in Machine learning

Download history 382/week @ 2024-05-13 86/week @ 2024-05-20 11/week @ 2024-05-27 296/week @ 2024-06-03

290 downloads per month

GPL-3.0-only

8MB
677 lines

RayBNN_Neural

Neural Networks with Sparse Weights in Rust using GPUs, CPUs, and FPGAs via CUDA, OpenCL, and oneAPI

Install Arrayfire

Install the Arrayfire 3.9.0 binaries at https://arrayfire.com/binaries/

or build from source https://github.com/arrayfire/arrayfire/wiki/Getting-ArrayFire

Add to 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.2"
RayBNN_Graph = "2.0.1"
RayBNN_Sparse = "2.0.1"
RayBNN_Optimizer = "2.0.1"
RayBNN_Neural = "2.0.1"

List of Examples

Dependencies

~8.5MB
~161K SLoC