pytrees