← Search

Noam Rozen

2 accepted papers

2025

Jamba: Hybrid Transformer-Mamba Language Models

ICLR 2025poster

We present Jamba, a novel hybrid Transformer-Mamba mixture-of-experts (MoE) architecture. Jamba interleaves blocks of Transformer and Mamba layers, enjoying the benefits of both model families. MoE is added in some of these layers to increase model capacity while keeping active parameter usage manag…

Cited by 3SourcePDFScholar
2021

Moser Flow: Divergence-based Generative Modeling on Manifolds

NeurIPS 2021oral

We are interested in learning generative models for complex geometries described via manifolds, such as spheres, tori, and other implicit surfaces. Current extensions of existing (Euclidean) generative models are restricted to specific geometries and typically suffer from high computational costs.…