Which version of Python does this work with?

kingoflolz / mesh-transformer-jax

Model parallel transformers in JAX and Haiku

Apache License 2.0

6.26k stars 890 forks source link

Which version of Python does this work with? #253

Open chrisbward opened 1 year ago

AidanShipperley commented 1 year ago

I've gotten it to work with python 3.8.0. I believe I tried 3.9 and ran into several issues. 3.7 may work as well but I doubt 3.6 will. Remember to use this issue and not the requirements.txt in this repo or you WILL run into dependency issues.

SpookOrSpooky commented 1 year ago

@AidanShipperley Can you drop your requirements.txt? Having a bit of trouble with the poetry code they quoted.

Trying to get this running on Windows