Checks if Python libraries like PyTorch and JAX are able to leverage an Nvidia GPU by running a minimal benchmark.

1. Clone this repository. For example, using

   git clone https://github.com/evanphilip/test-pythonlibs-gpu.git

2. Go to test-pythonlibs-gpu folder and use a tool of your choice to run the Python scripts. For example,

   1. run it using uv

      uv run test_torch.py

   2. run it using hatch

      hatch run test_torch.py