We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 96f65af commit 8f2798eCopy full SHA for 8f2798e
keras/src/backend/__init__.py
@@ -39,7 +39,7 @@
39
from keras.src.backend.tensorflow.core import Variable as BackendVariable
40
elif backend() == "jax":
41
from keras.src.backend.jax import * # noqa: F403
42
- from keras.src.backend.jax import Variable as BackendVariable
+ from keras.src.backend.jax.core import Variable as BackendVariable
43
elif backend() == "torch":
44
from keras.src.backend.torch import * # noqa: F403
45
from keras.src.backend.torch.core import Variable as BackendVariable
0 commit comments