Neural network framework in Haskell
- Base. This package defines the abstract neural-network, and a extendable specification of layers.
- Backend-hmatrix. This package implements the full-connect layer and convolution layer based on hmatrix library. It has a simple and plain representation but some issues in both time and space efficiency.
- Backend-blashs. This package implements the full-connect layer and convolution layer based on blas-hs library. A imperative interface for manipulating dense vector and matrix is devised for better storage utilization.
For example, building the package with SIMD-128
stack build --flag neural-network-blashs:vec128
- with openblas flag true in the flags section, please install the openblas by the official package management.
- or else, install blas/lapack package.
- The vec128 flag for neural-network-blashs can be turned on, and many operations will utilize SIMD for better performance.
neural-network-blashs:
vec128: true
- The flags vec256 and vec512 cause segment-fault for the moment.