Benchmark scripts for comparing tutorials in PyTorch and JAX - View it on GitHub