[codex] Add JAX oracle witnesses and unified notes#19
Merged
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
Add the initial unified JAX/PyTorch oracle expansion on top of the existing
tensor-ad-oracleslayout.This PR:
abs/identityandexp/identitycases, includingjvp,vjp,linearization,transpose, andadjoint_checkcases/abs/identity.jsonlandcases/exp/identity.jsonlrecords withjax_refvalidate_schema.pyis a maintainer/CI integrity tool while published JSONL files already store materialized numeric payloadsWhy
The branch extends the current oracle database instead of splitting a separate JAX repository, while making the JAX convention explicit in both the machine-readable witnesses and the math-note corpus.
Validation
uv run python -m unittest tests.test_repo_config tests.test_scripts tests.test_materialize tests.test_pytorch_v1 tests.test_validator_encoding tests.test_db_replay tests.test_jax_v1 tests.test_runtime_jax tests.test_math_notes_structure tests.test_math_registry -vuv run python scripts/verify_cases.pyuv run python scripts/check_replay.pyuv run python scripts/check_regeneration.pyNotes
uv run python scripts/validate_schema.pywas measured separately on the current branch and completed successfully in about 6 minutes for 9580 records.