gpt-neox
by EleutherAI
Pythonpushed almost 2 years ago
An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries
AI summary
Language model trainer
Provides a framework for training large-scale language models on GPUs with advanced features and optimizations.
- stars
- 7K
- forks
- 1K
- watching
- 125
- awesome list
- 1
Featured in 1 awesome list
Each link jumps to the spot where the list mentions gpt-neox.