message: [XLA/Docs] Use tf-nightly version for the notebook to use jit_compile=True
diff: "##### Copyright 2019 The TensorFlow Authors."
  ]
  },
  + {
  + "cell_type": "code",
  + "execution_count": null,
  + "metadata": {
  + "id": "H0rKBV4uZacD"
  + },
  + "outputs": [],
  + "source": [
  + "# In TF2.4 jit_compile is called experimental_compile",
  + "!pip install tf-nightly"
  + ]
  + },
  {
  "cell_type": "code",
  "execution_count": 0,
  