3 ms·
>> uses pure python to do numerics Many orgs use Python+Numpy, and that can be made faster using Jax >> this is not jit compilation. I disagree. Jax jit use
by erwincoumans 5y ago
>> uses pure python to do numerics
Many orgs use Python+Numpy, and that can be made faster using Jax
>> this is not jit compilation.
I disagree. Jax jit uses XLA, and XLA is a JIT compiler. An XLA graph is created during the runtime of the host program, and JIT-compiled to native code for the CPU, GPU or TPU.