Learnable latent embeddings for joint behavioral and neural analysis - Official implementation of CEBRA - View it on GitHub