Add comment

keras_core/backend/__init__.py

@@ -6,7 +6,8 @@ import numpy as np
 from keras_core.backend.config import backend
 
 if backend() == "pytorch":
-    # TODO: comment explanation
+    # torch needs to be imported first, otherwise it will
+    # segfault upon import.
     import torch
 
 from keras_core.backend.common.keras_tensor import KerasTensor

keras_core/backend/pytorch/numpy.py

@@ -1,4 +1,5 @@
 import torch
+
 from keras_core.backend.pytorch.core import to_torch_dtype
 
 

requirements.txt

@@ -10,3 +10,4 @@ pandas
 absl-py
 requests
 h5py
+torch