Instructions to use KexuanShi/Megatron-LM with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- NeMo
How to use KexuanShi/Megatron-LM with NeMo:
# tag did not correspond to a valid NeMo domain.
- Notebooks
- Google Colab
- Kaggle
| # Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved. | |
| # syntax=docker/dockerfile:1.3-labs | |
| ARG FROM_IMAGE_NAME | |
| FROM ${FROM_IMAGE_NAME} as main | |
| RUN apt-get update && \ | |
| apt-get install -y --no-install-recommends gettext && \ | |
| apt-get clean && \ | |
| wget https://github.com/mikefarah/yq/releases/download/v4.44.1/yq_linux_amd64 -O /usr/local/bin/yq && \ | |
| chmod a+x /usr/local/bin/yq | |
| ##### For NVIDIANS only ##### | |
| FROM main as jet | |
| ARG JET_API_VERSION | |
| RUN --mount=type=secret,id=JET_INDEX_URLS \ | |
| JET_INDEX_URLS=$(cat /run/secrets/JET_INDEX_URLS) && \ | |
| pip install --no-cache-dir jet-api==$JET_API_VERSION "jet-client~=4.0" --upgrade $JET_INDEX_URLS | |
| ENV PATH="$PATH:/opt/jet/bin" | |
| ### | |