PyTorch implementation of Policy Similarity Embeddings from "Contrastive Behavioral Similarity Embeddings for Generalization in Reinforcement Learning"
Original Tensorflow implementation
Starting point for this implementation was the excellent PyTorch implementation of DrQv2.