Learnable latent embeddings for joint behavioural and neural analysis