pub fn parse_pytorch_model( file_path: &Path, ) -> Result<HashMap<String, TensorStats>>
Parse a PyTorch model file (.pth, .pt) and extract tensor information