TaskMet: Task-driven Metric Learning for Model Learning - Bansal, Dishank; Chen, Ricky T. Q.; Mukadam, Mustafa; Amos, Brandon (Advances in Neural Information Processing Systems, 2023)