openteamsinc
pypi/megatron-core
Risk Profile
Package
Source
Risk Profile
Maturity:
Unknown
Repo not found
Health:
Unknown
Repo not found
Legal:
Unknown
Repo not found
Security:
Unknown
Repo not found
Package
pypi
megatron-core
Version
0.14.0
Last Release Date
about 2 months ago
License
Apache
Dependencies
torch
numpy
<2.0.0
packaging
mlm:
flask-restful
sentencepiece
tiktoken
wandb
dev:
tqdm
einops
~=0.8
tensorstore
!=0.1.46, !=0.1.72, ~=0.1
nvtx
~=0.2
transformers
~=4.53
multi-storage-client
~=0.20
opentelemetry-api
~=1.33.1
setuptools
<80.0.0
mamba-ssm
~=2.2
causal-conv1d
~=1.5
nv-grouped-gemm
~=1.1
transformer-engine
[pytorch]
<2.8.0, >=2.6.0a0
nvidia-resiliency-ext
<0.5.0, >=0.4.0a0
nvidia-modelopt
[torch]
<0.34.0, >=0.33.0a0
megatron-energon
[av_decode]
~=6.0
flashinfer-python
onnxscript
lts:
tqdm
einops
tensorstore
!=0.1.46, !=0.1.72
nvtx
transformers
zarr
setuptools
<80.0.0
Source
No source
