Skip to content

Neural Network primitives with multiple backends

License

Notifications You must be signed in to change notification settings

elijahmathews/NNlib.jl

 
 

Repository files navigation

NNlib

Build Status Build status Coverage

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, CuArrays provides GPU kernels, and Flux provides automatic differentiation; both can work together without explicitly being aware of each other.

About

Neural Network primitives with multiple backends

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Julia 100.0%