FluxML / Nnlib.jl
Licence: other
Neural Network primitives with multiple backends
Stars: ✭ 93
Programming Languages
julia
2034 projects
NNlib
This package will provide a library of functions useful for ML, such as softmax, sigmoid, convolutions and pooling. It doesn't provide any other "high-level" functionality like layers or AD.
Other packages can build on these functions as if they were defined in Base Julia; for example, CUDA.jl provides GPU kernels, and Zygote.jl provides automatic differentiation; both can work together without explicitly being aware of each other.
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].