An Adaptive Transfer data jax Library for NVIDIA GPUs
used for pmwd(GPUS)
export CMAKE_PREFIX_PATH=/opt/nvidia/hpc_sdk/Linux_x86_64/23.3/cmake python setup.py build
pip install . (or pip install -e .)
-
Notifications
You must be signed in to change notification settings - Fork 0
hai4john/jaxNvtransfer
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
An Adaptive Transfer data jax Library for NVIDIA GPUs
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published