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:9d1dfdc02ad7b778eec2c319226e691dc3710e56abc0abc1d43008ee1dc2f21c	RUN	`NEMO_VERSION=1.3.0 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh && chmod +x start-jupyter.sh`	09/03/2021 10:45 PM UTC
sha256:5ab55d3a5ab264e1daea010ac85bedf3630910c8ebca608a197880531a69a404	COPY	`tutorials /workspace/nemo/tutorials`	09/03/2021 10:45 PM UTC
sha256:e81081f3e13f056bf353a9cc447f4b045a9598b49167907e79d464180bc5e20f	COPY	`tests /workspace/nemo/tests`	09/03/2021 10:45 PM UTC
sha256:66f583cbd6cbe9591fb7e0e95560eb0cc7d3874dbac6cf2455149d1c3feefbf8	COPY	`examples /workspace/nemo/examples`	09/03/2021 10:45 PM UTC
sha256:85696c3196a72080f21ae99ff6e3ff9f52e413225a654bb4649742fd2e8bafe5	COPY	`scripts /workspace/nemo/scripts`	09/03/2021 10:45 PM UTC
sha256:162ece3c0c7813550682030c4dced5d8448277f8b345aa1e4c084f737d7e4c55	WORKDIR	`/workspace/nemo`	09/03/2021 10:45 PM UTC
sha256:5cd1d9dc936cdd335e0da2b3c8cad5b2e1edc815f1d72ecabca27fc270177183	RUN	`NEMO_VERSION=1.3.0 conda install -c numba numba=0.53.1`	09/03/2021 10:45 PM UTC
sha256:3f9ab7ec40ae680ab582f06294b2da8abd6fbb360dbc10390979cac7ed34bbf9	RUN	`NEMO_VERSION=1.3.0 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"`	09/03/2021 10:44 PM UTC
sha256:047db7e6c8bcd23af8625a2a531c613c4b3683c680586d6d56e57fde225dbac9	RUN	`NEMO_VERSION=1.3.0 /usr/bin/test -n "$NEMO_VERSION" && /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc && /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc`	09/03/2021 10:44 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ARG	`NEMO_VERSION=1.3.0`	09/03/2021 10:44 PM UTC