Megatron-DeepSpeed

Transformer trainer

Research tool for training large transformer language models at scale

Ongoing research training transformer language models at scale, including: BERT & GPT-2

GitHub

2k stars
25 watching
343 forks
Language: Python
last commit: almost 2 years ago
Linked from 1 awesome list


Backlinks from these awesome lists:

Related projects:

RepositoryDescriptionStars
bigscience-workshop/megatron-deepspeedA collection of tools and scripts for training large transformer language models at scale1,342
german-nlp-group/german-transformer-trainingTrains German transformer models to improve language understanding23
matlab-deep-learning/transformer-modelsAn implementation of deep learning transformer models in MATLAB209
openai/finetune-transformer-lmThis project provides code and model for improving language understanding through generative pre-training using a transformer-based architecture.2,167
fastnlp/cptA pre-trained transformer model for natural language understanding and generation tasks in Chinese482
jsksxs360/how-to-use-transformersA comprehensive guide to using the Transformers library for natural language processing tasks1,220
rdspring1/pytorch_gbw_lmTrains a large-scale PyTorch language model on the 1-Billion Word dataset123
huggingface/nanotronA pretraining framework for large language models using 3D parallelism and scalable training techniques1,332
jshilong/gpt4roiTraining and deploying large language models on computer vision tasks using region-of-interest inputs517
maxpumperla/elephasEnables distributed deep learning with Keras and Spark for scalable model training1,574
nlpodyssey/cybertronA Go package providing an easy interface to use pre-trained NLP models from the HuggingFace repository for tasks like text classification and machine translation.293
marella/ctransformersProvides a unified interface to various transformer models implemented in C/C++ using GGML library1,823
tongjilibo/bert4torchAn implementation of transformer models in PyTorch for natural language processing tasks1,257
gram-ai/radio-transformer-networksAn implementation of a machine learning-based communications system using deep learning techniques.127
pixart-alpha/pixart-sigmaDevelops a PyTorch model for 4K text-to-image generation using diffusion transformer1,711