t-SNE for Pytorch

Is there some workaround to do t-sne visualization of my autoencoder latent space in pytorch itself without using sklearn as it is relatively slow

You can use this implementation. It uses CUDA to speed things up. Or this one that allows multiprocessing. The speedup will be quite significant