Hello! Does anyone here know if the M3 or M2 will work better with Jax (deep learning library)? I’m currently debating which one to purchase! TIA - eviltoast