Hi,

Do you plan to add tracking of the PC's parameters by PyTorch? For example, a switch that would change the requires_grad of the PC's parameters from False to True. I would like to use the optimizers from torch.optim to train your PCs.

Best regards,
Milan