Skip testing JAX example for Jetson (#5012)

* skipping JAX example test

Author: nv-kmcgill53

qa/L0_backend_python/examples/test.sh

@@ -43,8 +43,11 @@ if [ "$TEST_JETSON" == "0" ]; then
     pip3 uninstall -y torch
     pip3 install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html
 fi
+
 # Install JAX
-pip3 install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+if [ "$TEST_JETSON" == "0" ]; then
+    pip3 install --upgrade "jax[cuda]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+fi
 
 git clone https://github.com/triton-inference-server/python_backend -b $PYTHON_BACKEND_REPO_TAG
 cd python_backend
@@ -112,35 +115,37 @@ wait $SERVER_PID
 # Example 3
 
 # JAX AddSub
-CLIENT_LOG="./jax_client.log"
-mkdir -p models/jax/1/
-cp examples/jax/model.py models/jax/1/model.py
-cp examples/jax/config.pbtxt models/jax/config.pbtxt
-run_server
-if [ "$SERVER_PID" == "0" ]; then
-    echo -e "\n***\n*** Failed to start $SERVER\n***"
-    cat $SERVER_LOG
-    RET=1
-fi
-
-set +e
-python3 examples/jax/client.py > $CLIENT_LOG
-if [ $? -ne 0 ]; then
-    echo -e "\n***\n*** Failed to verify jax example. \n***"
-    RET=1
-fi
+# JAX is not supported on Jetson
+if [ "$TEST_JETSON" == "0" ]; then
+    CLIENT_LOG="./jax_client.log"
+    mkdir -p models/jax/1/
+    cp examples/jax/model.py models/jax/1/model.py
+    cp examples/jax/config.pbtxt models/jax/config.pbtxt
+    run_server
+    if [ "$SERVER_PID" == "0" ]; then
+        echo -e "\n***\n*** Failed to start $SERVER\n***"
+        cat $SERVER_LOG
+        RET=1
+    fi
 
-grep "PASS" $CLIENT_LOG
-if [ $? -ne 0 ]; then
-    echo -e "\n***\n*** Failed to verify jax example. \n***"
-    cat $CLIENT_LOG
-    RET=1
-fi
-set -e
+    set +e
+    python3 examples/jax/client.py > $CLIENT_LOG
+    if [ $? -ne 0 ]; then
+        echo -e "\n***\n*** Failed to verify jax example. \n***"
+        RET=1
+    fi
 
-kill $SERVER_PID
-wait $SERVER_PID
+    grep "PASS" $CLIENT_LOG
+    if [ $? -ne 0 ]; then
+        echo -e "\n***\n*** Failed to verify jax example. \n***"
+        cat $CLIENT_LOG
+        RET=1
+    fi
+    set -e
 
+    kill $SERVER_PID
+    wait $SERVER_PID
+fi
 
 # Example 4