bcdarwin / NNlib.jl

Neural Network primitives with multiple backends

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

License:Other


Languages

Language:Julia 100.0%