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.16.0
Last Release Date
7 days ago
License
Apache
Dependencies
torch
>=2.6.0
numpy
packaging
>=24.2
mlm:
flask-restful
sentencepiece
tiktoken
wandb
transformers
dev:
nvidia-modelopt
[torch]
transformer-engine
[core_cu13,pytorch]
<2.12.0, >=2.9.0a0
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
nv-grouped-gemm
~=1.1
megatron-energon
[av_decode]
~=6.0
av
flashinfer-python
~=0.5.0
wget
onnxscript
fastapi
~=0.50
datasets
emerging_optimizers
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
nv-grouped-gemm
~=1.1
megatron-energon
[av_decode]
~=6.0
av
flashinfer-python
~=0.5.0
wget
onnxscript
fastapi
~=0.50
datasets
emerging_optimizers
Source
No source
