- added support for `torch>=2.0.0` and `torchvision=0.15.*` (fixes issue 165)
- implemented CKA in both NumPy and PyTorch (PyTorch runs 2x faster than NumPy)
- added CUDA support for CKA in PyTorch (100x faster than CKA in NumPy)
- added new function called `get_cka` where the user can set the `backend` (`torch` or `numpy`); function returns the CKA object
- added an unbiased CKA variant (from [here](https://github.com/google-research/google-research/blob/master/representation_similarity/Demo.ipynb)) in addition to vanilla CKA