fix: support ITensor amex in quantize and add `export_torch_mode()` wrapper (#4431)