Skip to content
OpenTrain AIFor AI Companies

Scaling Stick-Breaking Attention: An Efficient Implementation and In-depth Study

Published Oct 1, 2024
arXiv PDF
Researcher verdict
Starting point
Use as implementation starting point
Benchmark evidence
Thin evidence
Verify before relying
Time to first repro
A few hours
Fast first run
Risk flags
0
None detected

Results and benchmarks

Freshness tier: cold
Scaling Stick-Breaking Attention: An Efficient Implementation and In-depth Study presents a transformer method.
Task Dataset Metric Value Source
Language modeling ARC Accuracy 67.6 paper-derived

Audit each benchmark finding before selecting an implementation path. Evidence refs map to the disclosure below.

Implementation

Best maintained implementation now

Recommended
Confidence: High
Reproducibility: Strong
ibm/dolomite-engine

LM engine is a library for pretraining/finetuning LLMs

189 stars · 31 forks · Last push Aug 10, 2026 · Apache-2.0 license

  • License
  • CI
  • Dependencies
  • Docker

Official implementation from Papers with Code · Repository link is mentioned in the paper metadata · Community adoption signal (189 stars)

Why this implementation
Confidence: high

ibm/dolomite-engine is the strongest maintained implementation based on ranking signals. CI workflows are present. License is declared (Apache-2.0).

Open ibm/dolomite-engine
Reproduction risks
  • No repository-level red flags were detected, but paper-specific preprocessing and hyperparameter details may still be under-specified.
  • Selected ibm/dolomite-engine as the strongest maintained implementation for new work.
  • Includes CI workflow signals.
  • Includes dependency/environment manifest signals.
  • Repository activity is within the last 24 months.

Compare implementation paths

Compare maintenance quality, reproducibility coverage, and evidence confidence before choosing a reproduction baseline.

ibm/dolomite-engine
best maintained
Maintenance
Active
Confidence
High
Reproducibility
Strong
Stars
189
Last push
Aug 10, 2026 (16d)

Official implementation from Papers with Code · Repository link is mentioned in the paper metadata

  • No tagged releases
  • No Docker setup
shawntan/stickbreaking-attention
historical official
Maintenance
Stale
Confidence
High
Reproducibility
Limited
Stars
63
Last push
Jul 1, 2025 (420d)

Official implementation from Papers with Code · Repository link is mentioned in the paper metadata

  • No push in 12+ months
  • No CI pipeline detected
  • No tagged releases

Reproduction readiness

Time to first repro: hours
Last checked: Aug 25, 2026

Ready to run

Ready to reproduce

  • Clone ibm/dolomite-engine and install dependencies from pyproject.toml.
  • CI pipeline detected, so automated tests are in place.
  • Last updated 16 days ago.
Open ibm/dolomite-engine

Quick start

git clone https://github.com/ibm/dolomite-engine.git
pip install -e .

Hugging Face artifacts

No direct paper-linked artifacts were found. Showing strongest curated related artifacts for faster exploration.

Models

Curated Related

Broaden model search

Transformer Language modeling Language modeling Transformer

Datasets

No trustworthy datasets matches right now.

Search datasets on Hugging Face

Spaces

No trustworthy spaces matches right now.

Search spaces on Hugging Face

Research context

Tasks

Language modeling

Methods

Transformer

Domains

None detected

Evaluation and human feedback data

Open this paper in HFEPX to review benchmark signals, evaluation modes, and human-feedback protocol context.

Open in HFEPX
Explore similar papers

Jump to Paper2Code search queries derived from this paper's research context.

Language modeling Transformer

Data includes links from Papers with Code ( CC-BY-SA-4.0 ).