COMET: Learning Cardinality Constrained Mixture of Experts with Trees and Local Search

Authors: Shibal Ibrahim, Wenyu Chen, Hussein Hazimeh, Natalia Ponomareva, Zhe Zhao, Rahul Mazumder