4 lines
64 B
Python
4 lines
64 B
Python
import jax
|
|
|
|
if __name__ == "__main__":
|
|
print(jax.devices())
|