Models that can be used to benchmark the performance of models built in JAX vs the same models built in PyTorch - View it on GitHub