update python version (#21517)

* update python version

* fix GPU tests

* onlu update python version for flax

* disable 3.11 tests on other backends

* fix bugs

Author: divyashreepathihalli

.devcontainer/devcontainer.json

@@ -25,4 +25,4 @@
     "features": {
         "ghcr.io/devcontainers/features/github-cli:1": {}
     }
-}
+}

.github/workflows/actions.yml

@@ -22,7 +22,7 @@ jobs:
         backend: [tensorflow, jax, torch, numpy, openvino]
         nnx_enabled: [false]
         include:
-          - python-version: '3.10'
+          - python-version: '3.11'
             backend: jax
             nnx_enabled: true
     name: ${{ matrix.backend == 'jax' && format('Run tests ({0}, {1}, nnx_enabled = {2})', matrix.python-version, matrix.backend, matrix.nnx_enabled) || format('Run tests ({0}, {1})', matrix.python-version, matrix.backend) }}

README.md

@@ -88,7 +88,7 @@ os.environ["KERAS_BACKEND"] = "jax"
 import keras
 ```
 
-**Note:** The backend must be configured before importing `keras`, and the backend cannot be changed after 
+**Note:** The backend must be configured before importing `keras`, and the backend cannot be changed after
 the package has been imported.
 
 **Note:** The OpenVINO backend is an inference-only backend, meaning it is designed only for running model