Shawn-Guo-CN/JAX-vs-PyTorch-on-GPU

This project aims to compare between the computing speed of JAX and PyTorch on a various models of Nvidia GPUs.

PythonGPL-3.0