Closed nickjbrowning closed 3 months ago
still need to update torch and jax API to reflect CUDA API change...
Here is a pre-built version of the code in this pull request: wheels.zip, you can install it locally by unzipping wheels.zip and using pip to install the file matching your system
wheels.zip
pip
still need to update torch and jax API to reflect CUDA API change...