Joint inference and input optimization in equilibrium networks Shaojie Bai Carnegie Mellon University Carnegie Mellon University J. Zico Kolter