NeMo is a toolkit for creating Conversational AI applications.
The toolkit comes with extendable collections of pre-built modules and ready-to-use models for Automatic Speech Recognition (ASR), Natual Language Processing (NLP) and Text-to-Speech (TTS).
Built for speed, NeMo can utilize NVIDIA's Tensor Cores and scale out training to multiple GPUs and multiple nodes.
- Python 3.6, 3.7 or 3.8
- Pytorch 1.7.1. WARNING: This version currently does not support Pytorch 1.8.0
- NVIDIA GPU for training
Version | Status | Description |
---|---|---|
Latest | Documentation of the latest (i.e. main) branch. | |
Next | Documentation of the next release (i.e. r1.0.0rc1). | |
Stable | Documentation of the stable (i.e. stable) branch. |
A great way to start with NeMo is by checking one of our tutorials.
FAQ can be found on NeMo's Discussions board. You are welcome to ask questions or start discussions there.
Use this installation mode if you want the latest released version.
Use this installation mode if you want the a version from particular GitHub branch (e.g main).
Use this installation mode if you are contributing to NeMo.
The easiest way to start training with NeMo is by using NeMo's container. It has all requirements and NeMo 1.0.0b3 already installed.
If you chose to work with main branch, we recommend using NVIDIA's PyTorch container version 20.11-py3 and then installing from GitHub.
Many example can be found under "Examples" folder.
We welcome community contributions! Please refer to the CONTRIBUTING.md CONTRIBUTING.md for the process.
NeMo is under Apache 2.0 license.