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.15.2
Last Release Date
11 days ago
License
Apache
Dependencies
dev:
flashinfer-python
wget
onnxscript
nvidia-modelopt
[torch]
transformer-engine
[pytorch]
<2.10.0, >=2.9.0a0
nvidia-resiliency-ext
<0.5.0, >=0.4.0a0
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
setuptools
<80.0.0
mamba-ssm
~=2.2
causal-conv1d
~=1.5
nv-grouped-gemm
~=1.1
megatron-energon
[av_decode]
~=6.0
av
<16.0.0
lts:
tqdm
einops
tensorstore
!=0.1.46, !=0.1.72
nvtx
transformers
zarr
setuptools
<80.0.0
wget
torch
numpy
<2.0.0
packaging
>=24.2
mlm:
flask-restful
sentencepiece
tiktoken
wandb
transformers
Source
No source
