.. warning::
    TorchScript is deprecated, please use 
    `torch.export <https://docs.pytorch.org/tutorials/intermediate/torch_export_tutorial.html>`__ instead.