kingoflolz / mesh-transformer-jax

Model parallel transformers in JAX and Haiku

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Which version of Python does this work with?

chrisbward opened this issue · comments

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.

commented

@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