Remove redundant flatten/unflatten of PyTrees#1648

rauletorresc wants to merge 1 commit intomainPennyLaneAI/catalyst:mainfrom

raultorres/redundant_flatten_unflattenPennyLaneAI/catalyst:raultorres/redundant_flatten_unflattenCopy head branch name to clipboard