GitLab

Fixed deprecation warning for torch.utils._pytree._register_pytree_node in PyTorch 2.2
Co-authored-by: Yinghua <yzho0423@uni.sydney.edu.au>