Loosen Tensorflow version

[sarahmish]

Currently, Orion installation caps tensorflow at tensorflow<2.15, where the latest version is 2.18.

We should update the dependencies to support the latests versions.