fix : fix imports and renamings after refactoring #78

Job	Run time
tests (py3.11)	22s
tests (py3.12)	22s
tests (py3.13)	28s
import-independence (no TensorFlow)	22s
editable install	19s
	1m 53s

Provide feedback