flax
36ca83ac
- Install jaxlib+cuda if ACCELERATOR_TYPE =~ nvidia-
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
Install jaxlib+cuda if ACCELERATOR_TYPE =~ nvidia-
Author
andsteing
Parents
e8e2a986
Loading