|
|
|
|
|
|
by erwincoumans
1610 days 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. |
|
|