Non-AVX build of JAX for older CPUs

ins-amu / vbjax

A nascent Jax-based package for virtual brain modeling.

Apache License 2.0

7 stars 2 forks source link

Closed maedoc closed 1 year ago

maedoc commented 1 year ago

Our cluster has a bunch of Westmere nodes that don't work with AVX in the standard Jax wheels so we need ot build it from source :'(

maedoc commented 1 year ago

This may be relevant for newer x86_64 systems which are AMD and emulate AVX instructions?