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:5a09f4d2ee1a4eb82754bb8dd48037ca4297639b7a33a0ae4fa8138d5ad601a0	RUN	`NEMO_VERSION=1.0.2 printf "#!/bin/bash\njupyter lab --no-browser --allow-root --ip=0.0.0.0" >> start-jupyter.sh && chmod +x start-jupyter.sh`	07/30/2021 10:51 PM UTC
sha256:9725c4d84a8fa2051bdef2e2f733dce78892c37f878147a8cf6411904a80ab8a	COPY	`tutorials /workspace/nemo/tutorials`	07/30/2021 10:51 PM UTC
sha256:1953274fd4677fa1b9c074fa14853f982e3fe59a28f8060c2d1dcc7ead0ced53	COPY	`tests /workspace/nemo/tests`	07/30/2021 10:51 PM UTC
sha256:af6c1b1a92fe6dffbb2c055417264bd4758209c8f43d5990964b00cb42d52335	COPY	`examples /workspace/nemo/examples`	07/30/2021 10:51 PM UTC
sha256:ffe5b10cc4fca2d2abf3581b0e58be98d10e7c771bb938abadbd477db7a30506	COPY	`scripts /workspace/nemo/scripts`	07/30/2021 10:51 PM UTC
sha256:7fb37d981be9015ca2ec52277497c49b1a64981c4df7dc6e9dc7a45fd9a2d656	WORKDIR	`/workspace/nemo`	07/30/2021 10:51 PM UTC
sha256:2ecfceaa1e04eef472fe27089e4b464d02cf5065ff233a0ad21148fb8a148c35	RUN	`NEMO_VERSION=1.0.2 conda install -c numba numba=0.53.1`	07/30/2021 10:51 PM UTC
sha256:4bc121e1249d768fdfb16ad34e778988bff7c1965aa719a041b359ccc8d13012	RUN	`NEMO_VERSION=1.0.2 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"`	07/30/2021 10:50 PM UTC
sha256:3cf82fcbbc30bba6b4062c57c253da386bb582b490626e8d312aa2fac559c04f	RUN	`NEMO_VERSION=1.0.2 /usr/bin/test -n "$NEMO_VERSION" && /bin/echo "export NEMO_VERSION=${NEMO_VERSION}" >> /root/.bashrc && /bin/echo "export BASE_IMAGE=${BASE_IMAGE}" >> /root/.bashrc`	07/30/2021 10:50 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4	ARG	`NEMO_VERSION=1.0.2`	07/30/2021 10:50 PM UTC