Could we add a citation suggestion, maybe either in the `README` or as a file [like in e.g. `pytorch`](https://github.com/pytorch/pytorch/blob/master/CITATION)?