can the new released 0.3.1 version run on CUDA12 devices?

[fred2019-dev]

I modified the setup.py changing the jax[cuda13] to jax[cuda12], installation seems to be ok, but will trigger segementation fault, when running the flow-sphere example with compute_end=jax. So is cuda13 compulsory for 0.3.1??
######################################
I tried to simply replace the old xlb/ folder with the new 0.3.1 one, while still use pip install xlb[cuda] to start the environment, and it seems to work well for the originally failed collision_type=SmagorinskyLESBGK cases.

[mehdiataei]

Yes you can. I suggest installing from source and simply replace Warp/CUDA with cuda12 versions manually. Follow the instructions on JAX/Warp repo (ours could be outdated as they evolve rapidly). Glad that SmagorinskyLESBGK is working.

[fred2019-dev]

thanks for your answer, that is quite helpful