Custom runtime for JAX and other GPU dependencies
Base Runtime Python 3.9, Workbench, Cuda
Python packages installed nvidia-cuda-nvcc-cu112022-5.4, jax 0.3.25 and nvidia-pyindex 1.0.9
-
Notifications
You must be signed in to change notification settings - Fork 0
License
whiz/jax
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published