achiefa / NTK-interface

GNU General Public License v3.0
0 stars 0 forks source link

Implement higher derivatives of the NN #13

Closed achiefa closed 3 months ago

achiefa commented 6 months ago

This PR introduces the machinery needed to compute higher derivatives of the NN up to the third order. These are required in the calculation of the observables related to the Neural Tangent Hierarchy (NTH). The PR also introduce a jupyter notebook to analyse the observable and plot them as functions of the dimension of the network.

achiefa commented 5 months ago

At this point, there are few things to do:

achiefa commented 3 months ago

Will do that later. Time to merge...