Expressive density models for high energy physics using a custom latent space