flax
36ca83ac - Install jaxlib+cuda if ACCELERATOR_TYPE =~ nvidia-

Commit
5 years ago
Install jaxlib+cuda if ACCELERATOR_TYPE =~ nvidia-
Author
Parents
Loading