Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
test_gpu.py 78 Bytes
Newer Older
wangsen's avatar
readme.md  
wangsen committed
1
2
3
4 
import jax
import pymc as pm
print(jax.default_backend())
print(jax.devices())