Linear Probing tools

I would like to add some linear probing tools since it has been a widely used technique in evaluation of continual learning and non continual learning methods. I don't know what is the best way to do it. I think we could have it as an external layer that we add onto an existing representation, maybe with an attached method that trains it or an external method that trains it. I have to think about it.