suggestion: import的时候，改为相对引用

从推特过来的，你们这个项目，觉得非常棒🎉，期待更多的代码更新。
给个建议：megatron-lm最让我讨厌的一点，就是导入的时候，是绝对引用。
你们这个包，还在发展中，我看在import的时候，是绝对引用，希望改为相对引用。这样方便更多的开发者在这个上面进行二次开发。

当然这个也是不痛不痒的一个点。

No specific files, tests, or entry points are named. Start by locating the package's absolute imports and reviewing how the project is run both as an installed package and during development; done means imports consistently use the intended relative form without breaking supported usage.

Written by the indexing model from the issue text.