Neural_tangent_kernel loading ...