Learning Task Informed Abstractions