diff --git a/README.rst b/README.rst index 1beef6783..a7c8f72de 100644 --- a/README.rst +++ b/README.rst @@ -1,122 +1,12 @@ -|status| |documentation| |codeql| |license| |pypi| |pyversion| |downloads| |black| - -.. |status| image:: http://www.repostatus.org/badges/latest/active.svg - :target: http://www.repostatus.org/#active - :alt: Project Status: Active – The project has reached a stable, usable state and is being actively developed. - -.. |documentation| image:: https://readthedocs.com/projects/nvidia-nemo/badge/?version=main - :alt: Documentation - :target: https://docs.nvidia.com/deeplearning/nemo/user-guide/docs/en/main/ - -.. |license| image:: https://img.shields.io/badge/License-Apache%202.0-brightgreen.svg - :target: https://github.com/NVIDIA/NeMo/blob/master/LICENSE - :alt: NeMo core license and license for collections in this repo - -.. |pypi| image:: https://badge.fury.io/py/nemo-toolkit.svg - :target: https://badge.fury.io/py/nemo-toolkit - :alt: Release version - -.. |pyversion| image:: https://img.shields.io/pypi/pyversions/nemo-toolkit.svg - :target: https://badge.fury.io/py/nemo-toolkit - :alt: Python version - -.. |downloads| image:: https://static.pepy.tech/personalized-badge/nemo-toolkit?period=total&units=international_system&left_color=grey&right_color=brightgreen&left_text=downloads - :target: https://pepy.tech/project/nemo-toolkit - :alt: PyPi total downloads - -.. |codeql| image:: https://github.com/nvidia/nemo/actions/workflows/codeql.yml/badge.svg?branch=main&event=push - :target: https://github.com/nvidia/nemo/actions/workflows/codeql.yml - :alt: CodeQL - -.. |black| image:: https://img.shields.io/badge/code%20style-black-000000.svg - :target: https://github.com/psf/black - :alt: Code style: black - -.. _main-readme: - -**NVIDIA NeMo Framework** -=============== - -Latest News ------------ - -- 2023/12/06 `New NVIDIA NeMo Framework Features and NVIDIA H200 `_ - -.. image:: https://github.com/sbhavani/TransformerEngine/blob/main/docs/examples/H200-NeMo-performance.png - :target: https://developer.nvidia.com/blog/new-nvidia-nemo-framework-features-and-nvidia-h200-supercharge-llm-training-performance-and-versatility - :alt: H200-NeMo-performance - :width: 600 - -NeMo Framework has been updated with state-of-the-art features, -such as FSDP, Mixture-of-Experts, and RLHF with TensorRT-LLM to provide speedups up to 4.2x for Llama-2 pre-training on H200. -**All of these features will be available in an upcoming release.** - - - Introduction ------------ NVIDIA NeMo Framework is a generative AI framework built for researchers and pytorch developers working on large language models (LLMs), multimodal models (MM), automatic speech recognition (ASR), and text-to-speech synthesis (TTS). -The primary objective of NeMo is to provide a scalable framework for researchers and developers from industry and academia -to more easily implement and design new generative AI models by being able to leverage existing code and pretrained models. - -For technical documentation, please see the `NeMo Framework User Guide `_. - -All NeMo models are trained with `Lightning `_ and -training is automatically scalable to 1000s of GPUs. - -When applicable, NeMo models take advantage of the latest possible distributed training techniques, -including parallelism strategies such as - -* data parallelism -* tensor parallelism -* pipeline model parallelism -* fully sharded data parallelism (FSDP) -* sequence parallelism -* context parallelism -* mixture-of-experts (MoE) - -and mixed precision training recipes with bfloat16 and FP8 training. - -NeMo's Transformer based LLM and Multimodal models leverage `NVIDIA Transformer Engine `_ for FP8 training on NVIDIA Hopper GPUs -and leverages `NVIDIA Megatron Core `_ for scaling transformer model training. - -NeMo LLMs can be aligned with state of the art methods such as SteerLM, DPO and Reinforcement Learning from Human Feedback (RLHF), -see `NVIDIA NeMo Aligner `_ for more details. - -NeMo LLM and Multimodal models can be deployed and optimized with `NVIDIA Inference Microservices (Early Access) `_. - -NeMo ASR and TTS models can be optimized for inference and deployed for production use-cases with `NVIDIA Riva `_. - -For scaling NeMo LLM and Multimodal training on Slurm clusters or public clouds, please see the `NVIDIA Framework Launcher `_. -The NeMo Framework launcher has extensive recipes, scripts, utilities, and documentation for training NeMo LLMs and Multimodal models and also has an `Autoconfigurator `_ -which can be used to find the optimal model parallel configuration for training on a specific cluster. -To get started quickly with the NeMo Framework Launcher, please see the `NeMo Framework Playbooks `_ -The NeMo Framework Launcher does not currently support ASR and TTS training but will soon. - -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 generate text or images, transcribe audio, and synthesize speech in just a few lines of code. - -We have extensive `tutorials `_ that -can be run on `Google Colab `_ or with our `NGC NeMo Framework Container. `_ -and we have `playbooks `_ for users that want to train NeMo models with the NeMo Framework Launcher. -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. - -Key Features ------------- - -* `Large Language Models `_ -* `Multimodal `_ -* `Automatic Speech Recognition `_ -* `Text to Speech `_ -* `Computer Vision `_ +This repo implements multi-softmax in ASR models. Requirements ------------ @@ -125,103 +15,12 @@ Requirements 2) Pytorch 1.13.1 or above 3) NVIDIA GPU, if you intend to do model training -Developer Documentation ------------------------ - -.. |main| image:: https://readthedocs.com/projects/nvidia-nemo/badge/?version=main - :alt: Documentation Status - :scale: 100% - :target: https://docs.nvidia.com/deeplearning/nemo/user-guide/docs/en/main/ - -.. |stable| image:: https://readthedocs.com/projects/nvidia-nemo/badge/?version=stable - :alt: Documentation Status - :scale: 100% - :target: https://docs.nvidia.com/deeplearning/nemo/user-guide/docs/en/stable/ - -+---------+-------------+------------------------------------------------------------------------------------------------------------------------------------------+ -| Version | Status | Description | -+=========+=============+==========================================================================================================================================+ -| Latest | |main| | `Documentation of the latest (i.e. main) branch. `_ | -+---------+-------------+------------------------------------------------------------------------------------------------------------------------------------------+ -| Stable | |stable| | `Documentation of the stable (i.e. most recent release) branch. `_ | -+---------+-------------+------------------------------------------------------------------------------------------------------------------------------------------+ - Getting help with NeMo ---------------------- FAQ can be found on NeMo's `Discussions board `_. You are welcome to ask questions or start discussions there. -Installation ------------- - -The NeMo Framework can be installed in a variety of ways, depending on your needs. Depending on the domain, you may find one of the following installation methods more suitable. - -* Conda / Pip - Refer to the `Conda <#conda>`_ and `Pip <#pip>`_ sections for installation instructions. - - * This is recommended for Automatic Speech Recognition (ASR) and Text-to-Speech (TTS) domains. - * When using a Nvidia PyTorch container as the base, this is the recommended installation method for all domains. - -* Docker - Refer to the `Docker containers <#docker-containers>`_ section for installation instructions. - - * This is recommended for Large Language Models (LLM), Multimodal and Vision domains. - * NeMo LLM & Multimodal Container - `nvcr.io/nvidia/nemo:24.01.01.framework` - * NeMo Speech Container - `nvcr.io/nvidia/nemo:24.01.speech` - -Conda -~~~~~ - -We recommend installing NeMo in a fresh Conda environment. - -.. code-block:: bash - - conda create --name nemo python==3.10.12 - conda activate nemo - -Install PyTorch using their `configurator `_. - -.. code-block:: bash - - conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia - -The command used to install PyTorch may depend on your system. Please use the configurator linked above to find the right command for your system. - -Pip -~~~ -Use this installation mode if you want the latest released version. - -.. code-block:: bash - - apt-get update && apt-get install -y libsndfile1 ffmpeg - pip install Cython - pip install nemo_toolkit['all'] - -Depending on the shell used, you may need to use ``"nemo_toolkit[all]"`` instead in the above command. - -Pip (Domain Specific) -~~~~~~~~~~~~~~~~~~~~~ - -To install only a specific domain of NeMo, use the following commands. Note: It is required to install the above pre-requisites before installing a specific domain of NeMo. - -.. code-block:: bash - - pip install nemo_toolkit['asr'] - pip install nemo_toolkit['nlp'] - pip install nemo_toolkit['tts'] - pip install nemo_toolkit['vision'] - pip install nemo_toolkit['multimodal'] - -Pip from source -~~~~~~~~~~~~~~~ -Use this installation mode if you want the version from a particular GitHub branch (e.g main). - -.. code-block:: bash - - apt-get update && apt-get install -y libsndfile1 ffmpeg - pip install Cython - python -m pip install git+https://github.com/NVIDIA/NeMo.git@{BRANCH}#egg=nemo_toolkit[all] - - From source ~~~~~~~~~~~ Use this installation mode if you are contributing to NeMo. @@ -306,17 +105,7 @@ Note that RNNT requires numba to be installed from conda. Apex ~~~~ -NeMo LLM Domain training requires NVIDIA Apex to be installed. -Install it manually if not using the NVIDIA PyTorch container. - -To install Apex, run - -.. code-block:: bash - - git clone https://github.com/NVIDIA/apex.git - cd apex - git checkout b496d85fb88a801d8e680872a12822de310951fd - pip install -v --no-build-isolation --disable-pip-version-check --no-cache-dir --config-settings "--build-option=--cpp_ext --cuda_ext --fast_layer_norm --distributed_adam --deprecated_fused_adam" ./ +To install Apex, please follow the following URL: https://github.com/NVIDIA/apex.git It is highly recommended to use the NVIDIA PyTorch or NeMo container if having issues installing Apex or any other dependencies. @@ -337,87 +126,7 @@ packaging is also needed: With the latest versions of Apex, the `pyproject.toml` file in Apex may need to be deleted in order to install locally. - -Transformer Engine -~~~~~~~~~~~~~~~~~~ -NeMo LLM Domain 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. - -.. code-block:: bash - - pip install --upgrade git+https://github.com/NVIDIA/TransformerEngine.git@stable - -It is highly recommended to use the NVIDIA PyTorch or NeMo container if having issues installing Transformer Engine or any other dependencies. - -Transformer Engine requires PyTorch to be built with CUDA 11.8. - - -Flash Attention -~~~~~~~~~~~~~~~ -When traning Large Language Models in NeMo, users may opt to use Flash Attention for efficient training. Transformer Engine already supports Flash Attention for GPT models. If you want to use Flash Attention for non-causal models, please install `flash-attn `_. If you want to use Flash Attention with attention bias (introduced from position encoding, e.g. Alibi), please also install triton pinned version following the `implementation `_. - -.. code-block:: bash - - pip install flash-attn - pip install triton==2.0.0.dev20221202 - -NLP inference UI -~~~~~~~~~~~~~~~~~~~~ -To launch the inference web UI server, please install the gradio `gradio `_. - -.. code-block:: bash - - pip install gradio==3.34.0 - -NeMo Text Processing -~~~~~~~~~~~~~~~~~~~~ -NeMo Text Processing, specifically (Inverse) Text Normalization, is now a separate repository `https://github.com/NVIDIA/NeMo-text-processing `_. - -Docker containers -~~~~~~~~~~~~~~~~~ -We release NeMo containers alongside NeMo releases. For example, NeMo ``r1.23.0`` comes with container ``nemo:24.01.speech``, you may find more details about released containers in `releases page `_. - -To use built container, please run - -.. code-block:: bash - - docker pull nvcr.io/nvidia/nemo:24.01.speech - -To build a nemo container with Dockerfile from a branch, please run - -.. code-block:: bash - - DOCKER_BUILDKIT=1 docker build -f Dockerfile -t nemo:latest . - - -If you choose to work with the main branch, we recommend using NVIDIA's PyTorch container version 23.10-py3 and then installing from GitHub. - -.. code-block:: bash - - docker run --gpus all -it --rm -v :/NeMo --shm-size=8g \ - -p 8888:8888 -p 6006:6006 --ulimit memlock=-1 --ulimit \ - stack=67108864 --device=/dev/snd nvcr.io/nvidia/pytorch:23.10-py3 - Examples -------- Many examples can be found under the `"Examples" `_ folder. - - -Contributing ------------- - -We welcome community contributions! Please refer to `CONTRIBUTING.md `_ for the process. - -Publications ------------- - -We provide an ever-growing list of `publications `_ that utilize the NeMo framework. - -If you would like to add your own article to the list, you are welcome to do so via a pull request to this repository's ``gh-pages-src`` branch. -Please refer to the instructions in the `README of that branch `_. - -License -------- -NeMo is released under an `Apache 2.0 license `_.