#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

6 releases (2 stable)

2.0.3 Jun 19, 2024
2.0.1 Jun 5, 2024
0.1.3 Jun 5, 2024
0.1.2 May 19, 2024

#386 in Machine learning

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

443 downloads per month

GPL-3.0-only

44KB
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.3"
RayBNN_Graph = "2.0.3"
RayBNN_Sparse = "2.0.2"
RayBNN_Optimizer = "2.0.1"
RayBNN_Neural = "2.0.3"

List of Examples

Dependencies

~8.5MB
~163K SLoC