GitHub repository: matthewfeickert/nvidia-gpu-ml-library-test
Primary language: Python
Created: 2021-01-04T07:33:14Z
Description: Simple tests for JAX, PyTorch, and TensorFlow to test if the installed NVIDIA drivers are being properly picked up