Hi @yzhliu,
No, at this time the plan is to support tracing only. This stems from a desire to reduce the complexity of a) the front-end b) the represenatation and c) transforms and to keep FX accessible to a more general audience. True, this limits the type of models that can be transformed with FX, however we believe that the types of models folks work with (vision, NLP, etc) tend not to require sophisticated constructs