NeMo Framework Megatron Backend

NVIDIA

Container

NVIDIA

NeMo Framework Megatron Backend

NVIDIA NeMo™ framework Megatron backend supports pre-training, post-training, and reinforcement learning of LLMs and multi-modal generative AI models with state-of-the-art data processing, model training techniques, and flexible deployment options.

NVIDIA AI Enterprise Supported

Layer	Label		Created
sha256:2af37c06a86a5f9c5bf5ddcf063c1a006a39665b5ce9b49b7b89015905e4e85a	RUN	`NEMO_VERSION=1.0.1 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh && chmod +x start-jupyter.sh`	06/10/2021 12:56 AM UTC
sha256:d1cad126274859e0014f3aa049e2b09a5fb7b72156358fd25ce12723b6bc34e5	COPY	`tutorials /workspace/nemo/tutorials`	06/10/2021 12:56 AM UTC
sha256:ce3ad68ef70eee3abb02d7bf1159ed613b9472ed4fdd3be4d1ff3ef59df0d372	COPY	`tests /workspace/nemo/tests`	06/10/2021 12:56 AM UTC
sha256:cd39e801cd0bb48521590bea86e6d7ce2a72733aff57c065c258b17c5eecfcf5	COPY	`examples /workspace/nemo/examples`	06/10/2021 12:56 AM UTC
sha256:b47da202a6cfbe1edf1594a9e2a4db87786c4adbc6cf97ea0c928e75b3910aab	COPY	`scripts /workspace/nemo/scripts`	06/10/2021 12:56 AM UTC
sha256:cbb7eb53e0e29a44a6e73dcb79905d4146df3731035ef293ad007fd2cf436c37	WORKDIR	`/workspace/nemo`	06/10/2021 12:56 AM UTC
sha256:c3d864cd2c2ef26a046f920f3fc8ca4b7937cd72c70e5b95c3e0caaa45e8dcd1	RUN	`NEMO_VERSION=1.0.1 conda update -c numba numba -y && pip install --upgrade PyYAML`	06/10/2021 12:56 AM UTC
sha256:2315bb6bef8d3d6704a7d72a131d99c888c00ecd8192720f60497a25b2d2017e	RUN	`NEMO_VERSION=1.0.1 cd /tmp/nemo && pip install ".[all]" && python -c "import nemo.collections.asr as nemo_asr" && python -c "import nemo.collections.nlp as nemo_nlp" && python -c "import nemo.collections.tts as nemo_tts" && python -c "import nemo_text_processing.text_normalization as text_normalization"`	06/10/2021 12:55 AM UTC
sha256:b9877329d7775bba55041485ec9827faab7e335ba44b2e0e38c633a4248f36ba	RUN	`NEMO_VERSION=1.0.1 /usr/bin/test -n "$NEMO_VERSION" && /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc && /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc`	06/10/2021 12:55 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ARG	`NEMO_VERSION=1.0.1`	06/10/2021 12:55 AM UTC