Konsti enable rank4 ntk - Githubissues

zincware / ZnNL

A Python package for studying neural learning

Eclipse Public License 2.0

6 stars 1 forks source link

Konsti enable rank4 ntk #118

Closed KonstiNik closed 7 months ago

KonstiNik commented 7 months ago

[x] Make the Rank 4 NTK tensor default
[x] Allow the user to adjust the NTK calculation method in the neural-tangents
[x] Rescale trace recording to take into account rank 4 ntk