github.com GPUを認識していない場合には、以下のようなメッセージが出ます。 In [1]: import jax In [2]: jax.devices() An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu. Out[2]: [CpuDe…