docs
tests
torchzero
