
[megatron]
megatron-core>=0.12.0
transformers
safetensors
einops

[test]
pytest
