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.5.0
Required dependencies:
accelerate
|
comet-ml
|
datasets
|
diffusers
|
einops
|
flash-linear-attention
|
flashinfer-cubin
|
flashinfer-python
|
hydra-core
|
megatron-core
|
mistral-common
|
mlflow
|
nvidia-resiliency-ext
|
omegaconf
|
open-clip-torch
|
peft
|
pyyaml
|
qwen-vl-utils
|
regex
|
rich
|
six
|
tensorboard
|
timm
|
torch
|
tqdm
|
transformers
|
typing-extensions
|
wandb
Optional dependencies:
causal-conv1d
|
librosa
|
mamba-ssm
|
nemo-run
|
nvdlfw-inspect
|
pyarrow
|
transformer-engine
Downloads last day:
326
Downloads last week:
1,730
Downloads last month:
7,228