Port the model to PyTorch and compare performance
Port the model to PyTorch and compare performance