> does PyTorch have a similar concept of course https://github.com/pytorch/pytorch/blob/main/torch/utils/_pytree.py https://github.com/pytorch/pytorch/blob/mai

> does PyTorch have a similar concept of course https://github.com/pytorch/pytorch/blob/main/torch/utils/_pytree.py https://github.com/pytorch/pytorch/blob/main/torch/utils/_py...