Thanks for the idea, @ashutosh1919. Actually I have started using it in Colab. For the benefit of others who might be running in cloud environments where machine may or may not be linux, I am posting below a code snippet that checks if the machine is linux, and if it s it installs tfdf.
from sys import platform
if platform != "linux" and platform != "linux":
  print("'tensorflow_decision_forests' is currently only available for Linux.")

try:
  import tensorflow_decision_forests
except ModuleNotFoundError:
  !pip install tensorflow_decision_forests
  import tensorflow_decision_forests as tfdf