jax

aeff5b61 - Merge pull request #22080 from vfdev-5:add-device-kwarg-linspace-array