Detailed Dependency Setup Information

I am deeply interested in your team’s research, and I greatly admire the practical applications of your work in large language models. However, while attempting to reproduce your code using Python, I encountered numerous compatibility issues between dependencies. For instance, certain versions of JAX either too old or too recent tend to conflict with other required packages, making it quite difficult to set up the environment properly. Additionally, using newer versions of Python also results in errors.
Therefore, I would sincerely appreciate it if your team could provide a detailed list of dependencies and version requirements. This would be immensely helpful as a reference for academic reproduction and would serve as a valuable shortcut for researchers in the community.
Thank you very much!