Skip to content

support (old) jaxlib<=0.1.61

Vicentini Filippo requested to merge compat into master

Created by: PhilipVinc

mpi4jax does not support it, however it seems that later jaxlib versions require a cpu with AVX unless you build them yourself. It came up recently. It's low effort supporting it for now so we do it.

I think eventually this will get deleted but... yeah.

Merge request reports