Skip to content

Optimize pair-restricted and high-MOI JAX fitting - #58

Open
logan-blaine wants to merge 1 commit into
feature/cortado-successorfrom
codex/jax-cis-high-moi
Open

Optimize pair-restricted and high-MOI JAX fitting#58
logan-blaine wants to merge 1 commit into
feature/cortado-successorfrom
codex/jax-cis-high-moi

Conversation

@logan-blaine

Copy link
Copy Markdown
Collaborator

Summary

  • add exact pair-restricted fitting via --pairs-to-test
  • avoid unnecessary high-MOI guide-matrix retention for shared effects
  • compile minibatched SVI updates with lax.scan
  • assemble restricted beta parameters into dense matrices for fast high-MOI GEMMs
  • add coverage for pair restriction and high-MOI workflows

Validation

  • targeted pair/high-MOI/API tests pass
  • full non-fitted-workflow test suite passes
  • Gasperini TSS/NTC pilot completed successfully
  • Docker workflow dispatched separately on this branch

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