Skip to content
OpenTrain AIFor AI Companies

Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity

Published Jan 1, 2021
arXiv PDF
Researcher verdict
Starting point
Use as implementation starting point
Benchmark evidence
Missing
Not verified yet
Time to first repro
A few hours
Fast first run
Risk flags
0
None detected

Results and benchmarks

Freshness tier: cold
Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity presents a transformer method.

Implementation

Best maintained implementation now

Recommended
Confidence: Medium
Reproducibility: Strong
kyegomez/SwitchTransformers

Implementation of Switch Transformers from the paper: "Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsity"

146 stars · 17 forks · Last push Aug 3, 2026 · MIT license

  • License
  • CI
  • Dependencies
  • Docker

Matched via arXiv identifier search · Strong overlap with paper title keywords · Community adoption signal (146 stars)

Why this implementation
Confidence: medium

kyegomez/SwitchTransformers is the best available implementation candidate based on ranking signals, but recommendation confidence is not yet high. CI workflows are present. License is declared (MIT).

Open kyegomez/SwitchTransformers
Reproduction risks
  • No repository-level red flags were detected, but paper-specific preprocessing and hyperparameter details may still be under-specified.
  • Selected kyegomez/SwitchTransformers 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.

tensorflow/mesh
historical official
Maintenance
Archived
Confidence
High
Reproducibility
Limited
Stars
1,630
Last push
Nov 17, 2023 (1014d)

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

  • Repository archived
  • No push in 12+ months
  • No Docker setup
vikparuchuri/marker
alternative
Maintenance
Active
Confidence
Low
Reproducibility
Strong
Stars
39,275
Last push
Aug 25, 2026 (2d)

Community adoption signal (39275 stars)

  • No Docker setup
  • Low confidence match
efeslab/fiddler
alternative
Maintenance
Stale
Confidence
Low
Reproducibility
Moderate
Stars
267
Last push
Nov 18, 2024 (647d)

Community adoption signal (267 stars)

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

Reproduction readiness

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

Ready to run

Ready to reproduce

  • Clone kyegomez/SwitchTransformers and install dependencies from pyproject.toml.
  • CI pipeline detected, so automated tests are in place.
  • Last updated 24 days ago.
Open kyegomez/SwitchTransformers

Quick start

git clone https://github.com/kyegomez/SwitchTransformers.git
pip install -e .

Repositories and ecosystem

No additional verified repositories beyond the primary recommendation.

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 switch transformers scaling trillion

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

None detected

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.

Transformer

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