Skip to content

Fix torch regressions in PLNPCA and PLNnetwork#159

Open
EngineerDanny wants to merge 4 commits intoPLN-team:masterfrom
EngineerDanny:fix/torch-regressions
Open

Fix torch regressions in PLNPCA and PLNnetwork#159
EngineerDanny wants to merge 4 commits intoPLN-team:masterfrom
EngineerDanny:fix/torch-regressions

Conversation

@EngineerDanny
Copy link
Copy Markdown
Contributor

Summary

This PR fixes torch backend regressions in PLNPCA and PLNnetwork.

Changes

  • fix PLNPCA torch loglik_vec to match the existing rank backend and project() by adding a torch VE-step for projection
  • fix PLNnetwork torch fitting by passing only tensorizable data to the inherited torch optimizer
  • make fixed covariance torch Omega allocation pass device/dtype well
  • add targeted regression tests for PLNPCA and PLNnetwork torch paths

Validation

  • targeted testthat coverage added for the affected torch paths
  • targeted PLNPCA and PLNnetwork tests pass locally
  • full package test suite passes locally

Note

There are other PLN variants that dont support torch yet fully (PLNmixture, ZIPLN, ZIPLNnetwork), I just haven't used them in my analysis yet. So if I use them, then I will submit PR for them as well.

@EngineerDanny EngineerDanny force-pushed the fix/torch-regressions branch from a1337bf to e804dac Compare March 24, 2026 20:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant