[Wait] Remove legacy
Created by: PhilipVinc
We'll merge this after 3.0 is released.
I want to check how much it messes with the tests. legacy is used as a reference in some tests..
Trivia: This PR removes 11.4k LOC. NetKet will be left with ~13k LOC.