megatron-core (pypi): Megatron Core - a library for efficient and scalable training of transformer based models
Keywords: deep learning|machine learning|gpu|NLP|NLU|language|transformer|nvidia|pytorch|torch