Update requirements.txt

[sachinprasadhs]

Since Tensorflow and Tensorfow-cpu 2.19.0 is released, updating the version, also removed the version pin of JAX which was depending on the Tensorflow 2.19 release.

[sachinprasadhs]

Not sure why the tests are failing with memory error!

[fchollet]

Can you try only changing one of these things at a time to see which one is causing the test failure?

[fchollet]

It appears to be a cuDNN version issue?

E0000 00:00:1742492543.169566    2154 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered

[sachinprasadhs]

It appears to be a cuDNN version issue?

E0000 00:00:1742492543.169566    2154 cuda_dnn.cc:8579] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered

These warnings are there since long time with TensorFlow tensorflow/tensorflow#62075, I don't think that could be the issue.

[sampathweb]

@sachinprasadhs - I don't think we need both tensorflow-cpu and tensorflow in requirements.txt. May be they don't play well with each other anymore. Should we just keep cpu version for TF and Torch since that's what we have for jax?

[sachinprasadhs]

@sachinprasadhs - I don't think we need both tensorflow-cpu and tensorflow in requirements.txt. May be they don't play well with each other anymore. Should we just keep cpu version for TF and Torch since that's what we have for jax?

Made the changes, but any changes I make in this PR is resulting in memory error, not sure what is the reason.