Yanyang Wang

hi, I am a student of sysu, GPU cannot be used in this jax version, it is better to use tpu, and it is better to use video memory >=48G...

hi,What is your configuration version? Is it based on the information given by the author，I just used python3.6 and the configuration given by the author. Jax error will appear 

hi, I am a student of sysu, GPU cannot be used in this jax version, it is better to use tpu, and it is better to use video memory >=48G...