More code and corresponding instructions can be found at: https://github.com/a-quei/probe-neuralreg