NTK4A icon indicating copy to clipboard operation
NTK4A copied to clipboard

Implement BiRNN NTK Calculation (#1)

Open santacml opened this issue 4 years ago • 0 comments

This PR implements calculating the BiRNN NTK using summation and concatenation on the BiRNN hidden states.

The PR adds the python notebook, edits to utils, and NTK frob distance files for BiRNN.

The plots notebook is not edited to maintain consistency with the paper.

Merged via squash before creating PR.

Co-authored-by: Michael Santacroce (Microsoft Email)

santacml avatar Aug 26 '20 21:08 santacml