That does seem useful & cool. It would be great to have a Julia port since that would free us of python package management dependency hell. Since the license is Apache, a direct port seems like it might be the most straightforward way to go. I’m in the middle of other things now, but if anyone wants to give this a stab (manually, or using their favorite AI coding assistant to build the first draft) I’d be happy to review a package prototype.
I think the latter. Is there an up to date way of reading pth files in Julia that contain the trained model info? I assume some manual work would be required to build out the right neural network components to mirror the pytorch setup and then initialize it with the right data from the pth checkpoint.