Linear Probes Ai, linear_probe """Module for layer and neuron level linear-probe based analysis.