Skip to content

Neural Network primitives with multiple backends

License

Notifications You must be signed in to change notification settings

gabrielpreviato/NNlib.jl

 
 

Repository files navigation

NNlib.jl

Documentation CI Coverage

This package provides a library of functions useful for neural networks, such as softmax, sigmoid, batched multiplication, convolutions and pooling. Many of these are used by Flux.jl, which loads this package, but they may be used independently.

For use with automatic differentiation, this package defines gradients using ChainRules.jl. These will be seen by various packages including Zygote.jl.

To use these functions with CUDA.jl you will need NNlibCUDA.jl as well.

About

Neural Network primitives with multiple backends

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Julia 100.0%