From Graph Low-Rank Global Attention to 2-FWL Approximation. - Omri Puny; Heli Ben-Hamu; Yaron Lipman (CoRR, 2020)

Frame Averaging for Invariant and Equivariant Network Design. - Omri Puny; Matan Atzmon; Edward J. Smith; Ishan Misra; Aditya Grover; Heli Ben-Hamu; Yaron Lipman (ICLR, 2022)

Equivariant Polynomials for Graph Neural Networks. - Omri Puny; Derek Lim; Bobak Toussi Kiani; Haggai Maron; Yaron Lipman (ICML, 2023)

Mosaic-SDF for 3D Generative Models. - Lior Yariv; Omri Puny; Natalia Neverova; Oran Gafni; Yaron Lipman (CoRR, 2023)

Mosaic-SDF for 3D Generative Models. - Lior Yariv; Omri Puny; Oran Gafni; Yaron Lipman (CVPR, 2024)

D-Flow: Differentiating through Flows for Controlled Generation. - Heli Ben-Hamu; Omri Puny; Itai Gat; Brian Karrer; Uriel Singer; Yaron Lipman (ICML, 2024)

Puzzle: Distillation-Based NAS for Inference-Optimized LLMs. - Akhiad Bercovich; Tomer Ronen; Talor Abramovich; Nir Ailon; Nave Assaf; Mohammad Dabbah; Ido Galil; Amnon Geifman; Yonatan Geifman; Izhak Golan; Netanel Haber; Ehud Karpas; et al. (CoRR, 2024)

FFN Fusion: Rethinking Sequential Computation in Large Language Models. - Akhiad Bercovich; Mohammad Dabbah; Omri Puny; Ido Galil; Amnon Geifman; Yonatan Geifman; Izhak Golan; Ehud Karpas; Itay Levy; Zach Moshe; Najeeb Nabwani; Tomer Ronen; et al. (CoRR, 2025)

Llama-Nemotron: Efficient Reasoning Models. - Akhiad Bercovich; Itay Levy; Izik Golan; Mohammad Dabbah; Ran El-Yaniv; Omri Puny; Ido Galil; Zach Moshe; Tomer Ronen; Najeeb Nabwani; Ido Shahaf; Oren Tropp; et al. (CoRR, 2025)