Ramesh Sampath

Docker has env. variable LD_LIBRARY_PATH set but the path doesn't exist ``` tf-docker /tf/tensorflow > echo $LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 tf-docker /tf/tensorflow > ls /usr/local/ bin/ cuda-11.1/ etc/ include/ libexec/ sbin/ src/...

T5 backbone presets test are failing for Keras 2, but pass for Keras 3 (all backends). Log: https://btx-internal.corp.google.com/invocations/5135a970-cae4-4635-a2be-c9e3e8a73215/targets/keras_nlp%2Fgithub%2Fubuntu%2Fgpu%2Fkeras2%2Fpresubmit/log This test is now disabled for Keras 2 so the builds are...

Adds TF 2.16 Keras compatibility options.

Update CI to use TF 2.16 similar to Keras and Keras NLP.

Fix Segformer test for num_classes 1 with softmax

1

Keras has recently started to emit warnings when softmax has only one class. This PR fixes it by having more than one class at the output. Or we can re-compile...