Skip to content

Add JAX-based hyperedge network preprocessing design document#73

Merged
bibymaths merged 1 commit into
globalfrom
codex/create-design-document-for-jax-preprocessing-module
Jun 19, 2026
Merged

Add JAX-based hyperedge network preprocessing design document#73
bibymaths merged 1 commit into
globalfrom
codex/create-design-document-for-jax-preprocessing-module

Conversation

@bibymaths

Copy link
Copy Markdown
Owner

Motivation

  • Provide a concrete, developer-oriented blueprint for an independent, optional pure-JAX preprocessing module that discovers/prunes kinase–site–substrate hyperedges, detects 3-node motifs, builds a sparse theta tensor, and runs identifiability preprocessing prior to JAXopt fitting without changing existing default workflows.

Description

  • Add a single new design doc docs/jax_hyperedge_network_preprocessing.md that contains a repository scan summary, proposed module layout and public API, detailed JAX implementation plan (JIT/vmap/scan/batching boundaries and dtype guidance), algorithms for hyperedge discovery/pruning/motif detection/sparse-tensor construction/identifiability, integration points with networkmodel, output and plotting specifications, failure-mode safeguards, testing guidance, and a staged roadmap.

Testing

  • Performed automated repository scans to locate relevant integration points and ran git diff --check -- docs/jax_hyperedge_network_preprocessing.md to validate the new file format; the check completed without errors.

Codex Task

@bibymaths bibymaths merged commit 471b35e into global Jun 19, 2026
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant