Omri Puny

According to our database, Omri Puny authored at least 11 papers between 2020 and 2026.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of three.

2026

Extending Puzzle for Mixture-of-Experts Reasoning Models with Application to GPT-OSS Acceleration.

Akhiad Bercovich, Nir Ailon, Vladimir Anisimov, Tomer Asida, Nave Assaf, Mohammad Dabbah, Ido Galil, Amnon Geifman, Yonatan Geifman, Izhak Golan, Roi Koren, Itay Levy, Zach Moshe, Pavlo Molchanov, Najeeb Nabwani, Mostofa Patwari, Omri Puny, Tomer Ronen, Itamar Schen, Elad Segal, Ido Shahaf, Oren Tropp, Ran Zilberstein, Ran El-Yaniv

CoRR, February, 2026

2025

Space Group Conditional Flow Matching.

Omri Puny, Yaron Lipman, Benjamin Kurt Miller

CoRR, September, 2025

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, Itamar Schen, Elad Segal, Ido Shahaf, Oren Tropp, Ran Zilberstein, Ran El-Yaniv

CoRR, March, 2025

FFN Fusion: Rethinking Sequential Computation in Large Language Models.

Akhiad Bercovich, Mohammad Dabbah, Omri Puny, Ido Galil, Amnon Geifman, Yonatan Geifman, Izik Golan, Ehud Karpas, Itay Levy, Zach Moshe, Najeeb Nabwani, Tomer Ronen, Itamar Schen, Ido Shahaf, Oren Tropp, Ran Zilberstein, Ran El-Yaniv

Proceedings of the Advances in Neural Information Processing Systems 38: Annual Conference on Neural Information Processing Systems 2025, 2025

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, Roi Koren, Itay Levy, Pavlo Molchanov, Shahar Mor, Zach Moshe, Najeeb Nabwani, Omri Puny, Ran Rubin, Itamar Schen, Ido Shahaf, Oren Tropp, Omer Ullman Argov, Ran Zilberstein, Ran El-Yaniv

Proceedings of the Forty-second International Conference on Machine Learning, 2025

2024

D-Flow: Differentiating through Flows for Controlled Generation.

Heli Ben-Hamu, Omri Puny, Itai Gat, Brian Karrer, Uriel Singer, Yaron Lipman

Proceedings of the Forty-first International Conference on Machine Learning, 2024

Mosaic-SDF for 3D Generative Models.

Lior Yariv, Omri Puny, Oran Gafni, Yaron Lipman

Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2024

2023

Mosaic-SDF for 3D Generative Models.

Lior Yariv, Omri Puny, Natalia Neverova, Oran Gafni, Yaron Lipman

CoRR, 2023

Equivariant Polynomials for Graph Neural Networks.

Omri Puny, Derek Lim, Bobak Toussi Kiani, Haggai Maron, Yaron Lipman

Proceedings of the International Conference on Machine Learning, 2023

2022

Frame Averaging for Invariant and Equivariant Network Design.

Omri Puny, Matan Atzmon, Edward J. Smith, Ishan Misra, Aditya Grover, Heli Ben-Hamu, Yaron Lipman

Proceedings of the Tenth International Conference on Learning Representations, 2022

2020

From Graph Low-Rank Global Attention to 2-FWL Approximation.

Omri Puny, Heli Ben-Hamu, Yaron Lipman

CoRR, 2020