JaX is faster than Pytorch https://twitter.com/sanchitgandhi99/status/1649046661816287236 412 views18:45