EleutherAI / NeMo

NeMo: a toolkit for conversational AI

Home Page:https://nvidia.github.io/NeMo/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Project Status: Active – The project has reached a stable, usable state and is being actively developed. Documentation NeMo core license and license for collections in this repo Language grade: Python Total alerts Code style: black

NVIDIA NeMo

Introduction

NVIDIA NeMo is a conversational AI toolkit built for researchers working on automatic speech recognition (ASR), text-to-speech synthesis (TTS), large language models (LLMs), and natural language processing (NLP). The primary objective of NeMo is to help researchers from industry and academia to reuse prior work (code and pretrained models) and make it easier to create new conversational AI models.

All NeMo models are trained with Lightning and training is automatically scalable to 1000s of GPUs. Additionally, NeMo Megatron LLM models can be trained up to 1 trillion parameters using tensor and pipeline model parallelism. NeMo models can be optimized for inference and deployed for production use-cases with NVIDIA Riva.

Getting started with NeMo is simple. State of the Art pretrained NeMo models are freely available on HuggingFace Hub and NVIDIA NGC. These models can be used to transcribe audio, synthesize speech, or translate text in a just a few lines of code.

We have have extensive tutorials that can all be run on Google Colab.

For advanced users that want to train NeMo models from scratch or finetune existing NeMo models we have a full suite of example scripts that support multi-GPU/multi-node training.

Also see our introductory video for a high level overview of NeMo.

Key Features

Built for speed, NeMo can utilize NVIDIA's Tensor Cores and scale out training to multiple GPUs and multiple nodes.

Requirements

  1. Python 3.8 or above
  2. Pytorch 1.10.0 or above
  3. NVIDIA GPU for training

Documentation

Version Status Description
Latest Documentation Status Documentation of the latest (i.e. main) branch.
Stable Documentation Status Documentation of the stable (i.e. most recent release) branch.

Tutorials

A great way to start with NeMo is by checking one of our tutorials.

Getting help with NeMo

FAQ can be found on NeMo's Discussions board. You are welcome to ask questions or start discussions there.

Installation

Conda

We recommend installing NeMo in a fresh Conda environment.

Install PyTorch using their configurator.

Note

The command used to install PyTorch may depend on your system.

Pip

Use this installation mode if you want the latest released version.

Note

Depending on the shell used, you may need to use "nemo_toolkit[all]" instead in the above command.

Pip from source

Use this installation mode if you want the a version from particular GitHub branch (e.g main).

From source

Use this installation mode if you are contributing to NeMo.

Note

If you only want the toolkit without additional conda-based dependencies, you may replace reinstall.sh with pip install -e . when your PWD is the root of the NeMo repository.

RNNT

Note that RNNT requires numba to be installed from conda.

NeMo Megatron

NeMo Megatron training requires NVIDIA Apex to be installed. Install it manually if not using the NVIDIA PyTorch container.

Transformer Engine

NeMo Megatron GPT has been integrated with NVIDIA Transformer Engine Transformer Engine enables FP8 training on NVIDIA Hopper GPUs. Install it manually if not using the NVIDIA PyTorch container.

Note

Transformer Engine requires PyTorch to be built with CUDA 11.8.

NeMo Text Processing

NeMo Text Processing, specifically (Inverse) Text Normalization, requires Pynini to be installed.

Docker containers:

To build a nemo container with Dockerfile from a branch, please run

If you chose to work with main branch, we recommend using NVIDIA's PyTorch container version 22.11-py3 and then installing from GitHub.

Examples

Many examples can be found under "Examples" folder.

Contributing

We welcome community contributions! Please refer to the CONTRIBUTING.md CONTRIBUTING.md for the process.

Publications

We provide an ever growing list of publications that utilize the NeMo framework. Please refer to PUBLICATIONS.md. We welcome the addition of your own articles to this list !

Citation

License

NeMo is under Apache 2.0 license.

About

NeMo: a toolkit for conversational AI

https://nvidia.github.io/NeMo/

License:Apache License 2.0


Languages

Language:Python 56.5%Language:Jupyter Notebook 42.9%Language:Shell 0.3%Language:C++ 0.2%Language:HTML 0.1%Language:Dockerfile 0.0%Language:Groovy 0.0%Language:CSS 0.0%Language:Makefile 0.0%