jax

fc408d20 - Readded missing part to use locally built numpy in the build in tsan.yml