Examples failed about lightweight_mmm HOT 13 CLOSED

hey @blattnem ! Thanks for reporting this one.

I might need a bit more info on your enviroment.

1. What OS and version are you running?
2. What version of lightweight_mmm are you using?
3. Are you running the example as is? or with your own data?

Thanks!

from lightweight_mmm.

1. Mac OS Big Sur 11.5.2
2. lightweight_mmm 0.1.3
3. Run example as is

from lightweight_mmm.

The JAX support for Mac systems is not very extended, it actually depends on the underlying hardware to my understanding. For some ARM based CPU's might be good but I am not sure about sillicon based ones. Also this would be for CPU's but might be more tricky if you are trying GPU.

Any chance you have used JAX on that system for something else? Has that worked fine?

I think you might need to try to install and get working JAX separately and then go into lightweight_mmm.

My 2cents: Generally Linux would be a much less painful setup so if you have access to one I would recommend trying it there. Otherwise you can always use Colaboratory which is free and should give all you need.

from lightweight_mmm.

Hi Pablo
Sorry, same issue with linux (Manjaro linux).
I had to install python 3.8 to install everything
I get the same error as documented yesterday.
Best,
Marcel

from lightweight_mmm.

I install lightweight_mmm with pip

from lightweight_mmm.

Let me try to reproduce and get back to you.

When you said you had to install python 3.8, what version were you using before?

I have a couple setups with 3.7 and 3.9 where things seem to be running fine, but we definitely want to support 3.8 as well so I will look into it.

from lightweight_mmm.

I have python 3.10 on my linux box. Installed 3.8 because the dependency on tensorflow 2.7.2.

from lightweight_mmm.

This are my libs when running your example notebooks

jax 0.3.13
jaxlib 0.3.10
lightweight_mmm 0.1.3
numpyro 0.9.2
session_info 1.0.0

Click to view modules imported as dependencies

IPython 8.4.0
jupyter_client 7.3.4
jupyter_core 4.10.0
notebook 6.4.12

Python 3.8.13 (default, Jun 24 2022, 11:47:12) [GCC 11.2.0]
Linux-5.10.109-1-MANJARO-x86_64-with-glibc2.34

Session information updated at 2022-06-28 09:07

I got exactly the same error. I am not sure that it relates to python 3.8 version

from lightweight_mmm.

...and the same with python 3.9. I guess its a jax issue. What jax version are you using?

from lightweight_mmm.

Okay, thank you for that. Let me dig deeper into this one and get back to you.

from lightweight_mmm.

Looks like this is fixed by jax-0.3.14 & jaxlib-0.3.14 (I was having the same issue yesterday with the versions above)

from lightweight_mmm.

@usul83 good to know that upgrade worked! Thanks for pointing it out!

Im running our test with such versions again to confirm that is solved across the board.

@blattnem can you confirm it also solves your case?

from lightweight_mmm.

Hi there. Great news! Jax 0.3.14 does the job! Thanks.

from lightweight_mmm.