2 ms·Up and Running with Jax – JIT Compilation, Vectorizing Transformations and Auto2 points by todsacerdoti 2y ago