> You may be able to get the equivalent of jax via numba+numpy+autograd[1], but I haven't tried it before. I have tried it before [0]. And it works. As expecte

> You may be able to get the equivalent of jax via numba+numpy+autograd[1], but I haven't tried it before. I have tried it before [0]. And it works. As expected. Not much friction. [0]: sorry no github