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.17.0
Required dependencies:
numpy
|
nvidia-modelopt
|
packaging
|
torch
Optional dependencies:
accelerate
|
av
|
causal-conv1d
|
datasets
|
einops
|
emerging_optimizers
|
fastapi
|
flash-linear-attention
|
flashinfer-python
|
flask-restful
|
hypercorn
|
mamba-ssm
|
megatron-energon
|
multi-storage-client
|
nvidia-resiliency-ext
|
nvtx
|
onnxscript
|
openai
|
opentelemetry-api
|
orjson
|
quart
|
sentencepiece
|
tensorstore
|
tiktoken
|
tqdm
|
transformer-engine
|
transformers
|
wandb
|
wget
Downloads last day:
14,587
Downloads last week:
43,397
Downloads last month:
188,587