
[megatron]
megatron-core>=0.12.0
transformers
safetensors

[test]
pytest
