UER-py logo

UER-py

dbiir/UER-py

Open Source Pre-training Model Framework in PyTorch & Pre-trained Model Zoo

GraphCanon updated today · GitHub synced today

3.1k stars520 forksLast push 2y Python Apache-2.0

Decision brief

UER-py, an open-source PyTorch framework with a diverse model zoo for training and fine-tuning language models.

Good fit when

  • - When you need to work exclusively within the PyTorch ecosystem, UER-py provides extensive support for various pre-trained models and tasks without the necessity of switching frameworks.
  • - If your project requires access to a wide range of pre-trained models like BERT, RoBERTa, ALBERT, T5, GPT-2, or PEGASUS available in their model zoo.

Avoid when

  • - When you require more framework flexibility and are open to using TensorFlow or other deep learning libraries outside PyTorch.
  • - If your project is sensitive to maintenance updates but the UER-py repository has not seen recent active contribution, preferring a tool actively maintained might be better.
Pricing:
freemium - The framework itself is free and open-source under Apache 2.0 license providing flexibility for modification with no costs.
Requirements:
Min 8 GB RAM; - Requires Python environment setup; - Needs PyTorch installation

Observed Jul 12, 2026 · Source: enrich:decision_facts

Verify the decision

Maintenance and security

Full trust report
Maintenance
Dormant (836d since push)
As of today
Provenance
Not a fork · Organization account
As of today
Security (OSV)
No lockfile
As of 1mo

Public GitHub metadata and optional OSV scans. Signals, not a guarantee. Trust methodology.

Install

pip install UER-py
PyPI

Similar tools

Same-category neighbours. No typed graph edges are catalogued for this tool yet.

Evidence and technical details

Sourced facts, taxonomy, compatibility claims, README excerpt, and machine-readable endpoints.

Overview

UER-py is a comprehensive framework for pre-training models using PyTorch. It includes a variety of pre-trained models and supports tasks like classification, NER, and fine-tuning.

Capability facts

Languages
python

Source: github.language · Aug 23, 2026

Categories

Compatibility

Sourced claims from the README excerpt - not unsourced marketing copy.

Python runtimePython

Source: README excerpt (regex_v1, Aug 23, 2026)

* Python >= 3.6
Source link

Tags

README

Requirements

  • Python >= 3.6
  • torch >= 1.1
  • six >= 1.12.0
  • argparse
  • packaging
  • regex
  • For the pre-trained model conversion (related with TensorFlow) you will need TensorFlow
  • For the tokenization with sentencepiece model you will need SentencePiece
  • For developing a stacking model you will need LightGBM and BayesianOptimization
  • For the pre-training with whole word masking you will need word segmentation tool such as jieba
  • For the use of CRF in sequence labeling downstream task you will need pytorch-crf

For agents

This page has a .md twin and JSON over the API.

Was this helpful?

Anonymous feedback helps us improve pages and translations.