Thanks for creating this project! Would you please [add a license file](https://docs.github.com/en/repositories/managing-your-repositorys-settings-and-features/customizing-your-repository/licensing-a-repository) so your users would know how you allow to use it? Maybe you could consider using [the BSD-style license pytorch license](https://github.com/pytorch/pytorch/blob/main/LICENSE) or [using one of the other good alternatives](https://opensource.guide/legal/#which-open-source-license-is-appropriate-for-my-project)