actually, another question @tom
Was it supported in the past and the support was removed during recent changes?
I’m asking because you can see here (in cell 5) that accessing .saved_tensors from python while traversing used to work. The “orange blocks” were created that way.

  
      github.com
  
  
    szagoruyko/functional-zoo/blob/master/resnet-18-export.ipynb 8
{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "ResNet-18\n",
    "========="
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "[https://arxiv.org/abs/1512.03385](https://arxiv.org/abs/1512.03385)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},


  This file has been truncated. show original