PyPI page
Home page
Author:
None
License:
Apache 2.0
Summary:
Megatron Core - a library for efficient and scalable training of transformer based models
Latest version:
0.16.0
Required dependencies:
numpy
|
nvidia-modelopt
|
packaging
|
torch
Optional dependencies:
av
|
causal-conv1d
|
datasets
|
einops
|
emerging_optimizers
|
fastapi
|
flash-linear-attention
|
flashinfer-python
|
flask-restful
|
mamba-ssm
|
megatron-energon
|
multi-storage-client
|
nv-grouped-gemm
|
nvidia-resiliency-ext
|
nvtx
|
onnxscript
|
opentelemetry-api
|
sentencepiece
|
tensorstore
|
tiktoken
|
tqdm
|
transformer-engine
|
transformers
|
wandb
|
wget
Downloads last day:
9,488
Downloads last week:
57,715
Downloads last month:
206,909