← Search

Roi Cohen

4 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
2024

I Don't Know: Explicit Modeling of Uncertainty with an [IDK] Token

NeurIPS 2024poster

Large Language Models are known to capture real-world knowledge, allowing them to excel in many downstream tasks. Despite recent advances, these models are still prone to what are commonly known as hallucinations, causing them to emit unwanted and factually incorrect text. In this work, we propose a…