numpy torch dataclasses