The library versions on the “Nvidia-ProAI hardware” are older than the one used to train and save the state_dicts. PyTorch is backward compatible (you can load a checkpoint from an older PyTorch release in a newer version) but not forward compatible. Either update the libs on the Jetson or use an older release to store the checkpoint.
However, it’s still unclear if a randomly initialized model would also fail using the older library versions, so I would test it first.