Top 9 Python pre-training Projects

LLMSurvey

3 8,902 7.9 Python

The official GitHub page for the survey paper "A Survey of Large Language Models".

Project mention: Ask HN: Textbook Regarding LLMs | news.ycombinator.com | 2024-03-23

Here’s another one - it’s older but has some interesting charts and graphs.
https://arxiv.org/abs/2303.18223

Oscar

1 1,031 4.0 Python

Oscar and VinVL
InfluxDB

www.influxdata.com featured

Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
TencentPretrain

1 983 7.6 Python

Tencent Pre-training framework in PyTorch & Pre-trained Model Zoo
XPretrain

1 439 1.0 Python

Multi-modality pre-training

Project mention: CVPR 2024 Datasets and Benchmarks - Part 1: Datasets | dev.to | 2024-04-23

It was created by curating 3.8 million high-resolution videos from the publicly available HD-VILA-100M dataset. The dataset is created following these three steps:

STEP

1 306 4.0 Python

Code for our SIGKDD'22 paper Pre-training-Enhanced Spatial-Temporal Graph Neural Network For Multivariate Time Series Forecasting. (by zezhishao)
GearNet

1 244 4.4 Python

GearNet and Geometric Pretraining Methods for Protein Structure Representation Learning, ICLR'2023 (https://arxiv.org/abs/2203.06125)
Bamboo

3 161 4.1 Python

Bamboo: 4 times larger than ImageNet; 2 time larger than Object365; Built by active learning. (by ZhangYuanhan-AI)
SaaSHub

www.saashub.com featured

SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives
AMRBART

1 89 2.8 Python

Code for our paper "Graph Pre-training for AMR Parsing and Generation" in ACL2022
tsdae

1 3 5.1 Python

Tranformer-based Denoising AutoEncoder for Sentence Transformers Unsupervised pre-training.

Project mention: Tranformer-based Denoising AutoEncoder for ST Unsupervised pre-training | news.ycombinator.com | 2024-02-04

A new PyPI package for training sentence embedding models in just 2 lines.
The acquisition of sentence embeddings often necessitates a substantial volume of labeled data. However, in many cases and fields, labeled data is rarely accessible, and the procurement of such data is costly. In this project, we employ an unsupervised process grounded in pre-trained Transformers-based Sequential Denoising Auto-Encoder (TSDAE), introduced by the Ubiquitous Knowledge Processing Lab of Darmstadt, which can realize a performance level reaching 93.1% of in-domain supervised methodologies.
The TSDAE schema comprises two components: an encoder and a decoder. Throughout the training process, TSDAE translates tainted sentences into uniform-sized vectors, necessitating the decoder to reconstruct the original sentences utilizing this sentence embedding. For good reconstruction quality, the semantics must be captured well in the sentence embeddings from the encoder. Subsequently, during inference, the encoder is solely utilized to form sentence embeddings.
GitHub : https://github.com/louisbrulenaudet/tsdae
Installation :

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).

Index

What are some of the best open-source pre-training projects in Python? This list will help you:

	Project	Stars
1	LLMSurvey	8,902
2	Oscar	1,031
3	TencentPretrain	983
4	XPretrain	439
5	STEP	306
6	GearNet	244
7	Bamboo	161
8	AMRBART	89
9	tsdae	3