Skip to content
backprop primitives for hmatrix
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
src/Numeric/LinearAlgebra/Static
test
.gitignore
.travis.yml
CHANGELOG.md
LICENSE
README.md
Setup.hs
package.yaml
stack.yaml

README.md

hmatrix-backprop

hmatrix-backprop on Hackage Build Status

hmatrix operations lifted for backprop, along with orphan instances of Backprop.

Meant to act as a drop-in replacement to the API of Numeric.LinearAlgebra.Static. Just change your imports, and your functions are automatically backpropagatable. Useful types are all re-exported.

Formulas for gradients come from the following papers:

Some functions are not yet implemented! See module documentation for details. PR's definitely appreciated :)

Tests

Currently numeric tests are implemented as property tests using hedgehog, but it is possible that the answers might differ from the true values by an amount undetectable by property tests.

All functions currently are tested except for the higher-order functions.

They are tested by "nudging" components of inputs and checking if the change in the function outputs match what is expected from the backpropagated gradient.

TODO

  1. Now that new backprop no longer requires Num, we can lift normal hmatrix operations as well.
  2. Statically sized convolutions. Should probably add this to hmatrix instead first, though.
You can’t perform that action at this time.