Fix JAX convert_to_tensor dropping input dtype when floatx is bfloat16. - #23683

Fix JAX convert_to_tensor dropping input dtype when floatx is bfloat16.#23683

maitry63 wants to merge 3 commits into

keras-team:masterkeras-team/keras:masterfrom

maitry63:fix_convert-to-tensor_bfloat16_jaxmaitry63/keras:fix_convert-to-tensor_bfloat16_jaxCopy head branch name to clipboard