← Search

Itay Levy

3 accepted papers

2025

FFN Fusion: Rethinking Sequential Computation in Large Language Models

NeurIPS 2025spotlight

We introduce \textit{FFN Fusion}, an architectural optimization technique that reduces sequential computation in large language models by identifying and exploiting natural opportunities for parallelization. Our key insight is that sequences of Feed-Forward Network (FFN) layers, particularly those r…

Cited by 0SourceScholar
2025

Puzzle: Distillation-Based NAS for Inference-Optimized LLMs

ICML 2025poster

Large language models (LLMs) offer remarkable capabilities, yet their high inference costs restrict wider adoption. While increasing parameter counts improves accuracy, it also broadens the gap between state-of-the-art capabilities and practical deployability. We present **Puzzle**, a hardware-aware…

Cited by 2SourcePDFScholar
2023

Diverse Demonstrations Improve In-context Compositional Generalization

ACL 2023long

In-context learning has shown great success in i.i.d semantic parsing splits, where the training and test sets are drawn from the same distribution. In this setup, models are typically prompted with demonstrations that are similar to the input utterance. However, in the setup of compositional genera…