Optimize memory peak for _preprocess_state_vector
in LightningTensor
#279
Loading
_preprocess_state_vector
in LightningTensor
#279