4 ms·Side note: I like Pytorch but eager pytorch is not faster the jax.jit or tf.function codeby cgarciae 6y agoSide note: I like Pytorch but eager pytorch is not faster the jax.jit or tf.function code