Investigating jet physics through machine learning-enhanced representation