PyPI page
Home page
Author:
None
License:
Apache License Version 2.0, January 2004 http://www.apache.org/licenses/ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 1. Definitions. "License" shall mean the terms and conditions f...
Summary:
Megatron Bridge: Training Recipes for Megatron-based LLM and VLM models
Latest version:
0.4.1
Required dependencies:
accelerate
|
causal-conv1d
|
comet-ml
|
datasets
|
diffusers
|
einops
|
flash-linear-attention
|
hydra-core
|
imageio
|
imageio-ffmpeg
|
mamba-ssm
|
megatron-core
|
mlflow
|
nvidia-resiliency-ext
|
omegaconf
|
open-clip-torch
|
peft
|
pyyaml
|
qwen-vl-utils
|
regex
|
rich
|
six
|
tensorboard
|
timm
|
torch
|
tqdm
|
transformer-engine
|
transformers
|
typing-extensions
|
wandb
Optional dependencies:
nemo-run
|
nvdlfw-inspect
|
pyarrow
Downloads last day:
1,222
Downloads last week:
5,384
Downloads last month:
28,673