Skip to content
implementation starting point
Benchmarks: thin evidence
Time to repro: a few hours
1 risk flag
jax

Results & Benchmarks

Freshness tier: cold
Direct + Inferred Evidence
Natural language processing
CodeGen
6.1B
34
Source: paper fulltext
Natural language processing
Dimensions per head
350M
256
Source: paper fulltext

Benchmark evidence drill-down

2 findings

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

Task Dataset Metric Value Source Evidence refs
Natural language processing CodeGen 6.1B 34 paper-derived No explicit refs
Natural language processing Dimensions per head 350M 256 paper-derived No explicit refs

CodeGen: An Open Large Language Model for Code with Multi-Turn Program Synthesis is the primary contribution described in this paper.

Use This Implementation Because…

Confidence: high

salesforce/jaxformer is the strongest maintained implementation based on ranking signals. License is declared (BSD-3-Clause). Dependency/environment manifests are present.

Open salesforce/jaxformer

Reproduction Risks

  • No CI workflows detected
Evidence disclosure

Evidence graph: 4 refs, 4 links.

Utility signals: depth 90/100, grounding 95/100, status high.

Implementation Comparison

Top 3 paths

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

salesforce/CodeGen
historical official
Maintenance: Active
Confidence: High
Reproducibility: Limited

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

Stars
5,176
Last push
Jun 2, 2026 (16d ago)

Risk flags

  • No CI pipeline detected
  • No tagged releases
  • No Docker setup
salesforce/jaxformer
best maintained
Maintenance: Active
Confidence: High
Reproducibility: Moderate

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

Stars
299
Last push
Jun 2, 2026 (16d ago)
Dependencies

Risk flags

  • No CI pipeline detected
  • No tagged releases
  • No Docker setup
openlmlab/moss
alternative
Maintenance: Active
Confidence: Low
Reproducibility: Moderate

Partial overlap with paper title keywords · Community adoption signal (12140 stars)

Stars
12,140
Last push
May 27, 2026 (22d ago)
Dependencies

Risk flags

  • No CI pipeline detected
  • No tagged releases
  • No Docker setup

Best implementation now

salesforce/jaxformer
Confidence: High
Reproducibility: Moderate

Minimal library to train LLMs on TPU in JAX with pjit().

Stars: 299
Forks: 38
Last push: Jun 2, 2026
License: BSD-3-Clause
Official implementation from Papers with Code
Repository link is mentioned in the paper metadata
Community adoption signal (299 stars)
License ✓
CI –
Deps ✓
Docker –
  • Selected salesforce/jaxformer as the strongest maintained implementation for new work.
  • Includes dependency/environment manifest signals.
  • Repository activity is within the last 24 months.
  • Official repository is preserved separately as historical context.

Historical official implementation

Preserved for provenance. Not recommended as the default path for new builds.

salesforce/CodeGen
Stars: 5,176
Last push: Jun 2, 2026

Reproduction readiness

Setup Required
Time to first repro: hours
Last checked: Jun 17, 2026

Dependencies pinned, manual setup needed

  • · salesforce/jaxformer has requirements.txt but requires manual environment setup.
  • · No Dockerfile — you will set up the environment manually.
  • · No CI pipeline — test coverage is unknown.
Open salesforce/jaxformer

Quick start

git clone https://github.com/salesforce/jaxformer.git
pip install -r requirements.txt

Additional implementations

No additional verified repositories beyond the primary recommendation.

These repositories had low-confidence matching signals and are hidden by default.

Hugging Face artifacts

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

Models

Broaden model search

Transformer Natural language processing Transformer Natural Language Processing Natural language processing

Datasets

No trustworthy dataset matches right now.

Search datasets on Hugging Face

Spaces

No trustworthy demo spaces right now.

Search spaces on Hugging Face
Search models Search datasets Search spaces

Research context

Tasks

Natural language processing

Methods

Transformer

Domains

Natural Language Processing

Evaluation & 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.

Natural language processing Transformer Natural Language Processing

Need human evaluators for your AI research? Scale annotation with expert AI Trainers.

Post a Job Get a Quote