Pytorch용으로 유사한것도 있으려나요?