Ah yeah the size of the array may be too low and simd and threads are not composing well.
However we already beat JAX with both (KA or LV)
Ah yeah the size of the array may be too low and simd and threads are not composing well.
However we already beat JAX with both (KA or LV)