The recording allows for additional observables of the NTK besides Trace and NTK entropy.
The explicit calculation of the observables inside the recording will likely be removed later, but structural changes to enable the correct computation of the observables have been done in this PR.
Goal of this PR is to add more NTK observales.
The recording allows for additional observables of the NTK besides Trace and NTK entropy.
The explicit calculation of the observables inside the recording will likely be removed later, but structural changes to enable the correct computation of the observables have been done in this PR.