alpa-projects / alpa

Training and serving large-scale neural networks with auto parallelization.
https://alpa.ai
Apache License 2.0
3.07k stars 357 forks source link

Working Environment for Alpa? #975

Open man2machine opened 2 weeks ago

man2machine commented 2 weeks ago

Is there a working conda env.yml file that will work with Alpa? Most of the libraries used in this repo (jax, pytorch, tensorflow, transformers, etc.) have undergone major revisions and I have been spending more than a day just setting up a working conda environment, which is still not working.

It would be greatly beneficial if you can do this! Even just pasting the environment packages/versions that you use from conda list so I can replicate it exactly would be beneficial.