Puzzle: Distillation-Based NAS for Inference-Optimized LLMs - Akhiad Bercovich (arXiv.org)