openteamsinc
pypi/megatron-core
Risk Profile
Package
Source
Risk Profile
Maturity:
Mature
Health:
Healthy
Legal:
Caution Needed
License has additional text that may require review before use
License includes a patent grant clause
Security:
Healthy
Package
pypi
megatron-core
Version
0.17.0
Last Release Date
3 days ago
License
Apache
Dependencies
torch
>=2.6.0
numpy
packaging
>=24.2
training:
flask-restful
sentencepiece
tiktoken
wandb
transformers
accelerate
mlm:
flask-restful
sentencepiece
tiktoken
wandb
transformers
accelerate
dev:
nvidia-modelopt
[torch]
transformer-engine
[core_cu13,pytorch]
nvidia-resiliency-ext
tqdm
einops
~=0.8
tensorstore
!=0.1.46, !=0.1.72, ~=0.1
nvtx
~=0.2
multi-storage-client
~=0.27
opentelemetry-api
~=1.33.1
mamba-ssm
~=2.2
causal-conv1d
~=1.5
flash-linear-attention
~=0.4.0
megatron-energon
[av_decode]
~=6.0
av
flashinfer-python
~=0.5.0
wget
onnxscript
fastapi
~=0.50
datasets
emerging_optimizers
hypercorn
quart
openai
[aiohttp]
orjson
lts:
tqdm
einops
~=0.8
tensorstore
!=0.1.46, !=0.1.72, ~=0.1
nvtx
~=0.2
multi-storage-client
~=0.27
opentelemetry-api
~=1.33.1
mamba-ssm
~=2.2
causal-conv1d
~=1.5
megatron-energon
[av_decode]
~=6.0
av
flashinfer-python
~=0.5.0
wget
onnxscript
fastapi
~=0.50
datasets
emerging_optimizers
Source
Location
NVIDIA/Megatron-LM
Last Source Update
1 day ago
Licenses
Apache License 2.0
(text added)
Distribution Destinations
pypi/megatron-core
pypi/megatron-fsdp
