Versatile Energy-Based Models for High Energy Physics