{"record_type":"pith_number_record","schema_url":"https://pith.science/schemas/pith-number/v1.json","pith_number":"pith:2026:VMEOFFHDXINCQE2S3O3LSGLZO6","short_pith_number":"pith:VMEOFFHD","schema_version":"1.0","canonical_sha256":"ab08e294e3ba1a281352dbb6b9197977ac89cbc69084671a982360be98825511","source":{"kind":"arxiv","id":"2606.09001","version":1},"attestation_state":"computed","paper":{"title":"JAX-AMG: A GPU-Accelerated Differentiable Sparse Linear Solver Library for JAX","license":"http://creativecommons.org/licenses/by-sa/4.0/","headline":"","cross_cats":["physics.comp-ph"],"primary_cat":"cs.MS","authors_text":"Jian-Xun Wang, Xiantao Fan, Yi Liu","submitted_at":"2026-06-08T03:57:19Z","abstract_excerpt":"Sparse linear systems from PDE discretizations are central to scientific computing, yet no existing JAX-ecosystem solver simultaneously provides GPU-accelerated algebraic multigrid (AMG), automatic differentiation (AD), and distributed multi-GPU execution. JAX-AMG fills this gap by wrapping the Nvidia AmgX solver suite as a native JAX primitive, exposing AMG and Krylov methods with configurable preconditioners through a unified interface compatible with JIT compilation, reverse-mode AD via adjoint methods, batched solves, and MPI-based distributed execution. Solver caching amortizes setup cost"},"verification_status":{"content_addressed":true,"pith_receipt":true,"author_attested":false,"weak_author_claims":0,"strong_author_claims":0,"externally_anchored":false,"storage_verified":false,"citation_signatures":0,"replication_records":0,"graph_snapshot":true,"references_resolved":false,"formal_links_present":false},"canonical_record":{"source":{"id":"2606.09001","kind":"arxiv","version":1},"metadata":{"license":"http://creativecommons.org/licenses/by-sa/4.0/","primary_cat":"cs.MS","submitted_at":"2026-06-08T03:57:19Z","cross_cats_sorted":["physics.comp-ph"],"title_canon_sha256":"f7f1a95c78be850cea2fa39d93ce8646be9404580a34e0e974187a87def95edb","abstract_canon_sha256":"f9b2a44589653a6e4dbe8326b658e5c22cf1d3bc24c6b400c04776d8ae40c10b"},"schema_version":"1.0"},"receipt":{"kind":"pith_receipt","key_id":"pith-v1-2026-05","algorithm":"ed25519","signed_at":"2026-06-09T02:07:52.820636Z","signature_b64":"A/FC+2KOzQZ+U2qhHV/eBKbhOir0aEU/sQ4ahkxns7VDWpBMhDfbhl9eornmR9n8AZlu938fTyj62ojq310IAA==","signed_message":"canonical_sha256_bytes","builder_version":"pith-number-builder-2026-05-17-v1","receipt_version":"0.3","canonical_sha256":"ab08e294e3ba1a281352dbb6b9197977ac89cbc69084671a982360be98825511","last_reissued_at":"2026-06-09T02:07:52.819768Z","signature_status":"signed_v1","first_computed_at":"2026-06-09T02:07:52.819768Z","public_key_fingerprint":"8d4b5ee74e4693bcd1df2446408b0d54"},"graph_snapshot":{"paper":{"title":"JAX-AMG: A GPU-Accelerated Differentiable Sparse Linear Solver Library for JAX","license":"http://creativecommons.org/licenses/by-sa/4.0/","headline":"","cross_cats":["physics.comp-ph"],"primary_cat":"cs.MS","authors_text":"Jian-Xun Wang, Xiantao Fan, Yi Liu","submitted_at":"2026-06-08T03:57:19Z","abstract_excerpt":"Sparse linear systems from PDE discretizations are central to scientific computing, yet no existing JAX-ecosystem solver simultaneously provides GPU-accelerated algebraic multigrid (AMG), automatic differentiation (AD), and distributed multi-GPU execution. JAX-AMG fills this gap by wrapping the Nvidia AmgX solver suite as a native JAX primitive, exposing AMG and Krylov methods with configurable preconditioners through a unified interface compatible with JIT compilation, reverse-mode AD via adjoint methods, batched solves, and MPI-based distributed execution. Solver caching amortizes setup cost"},"claims":{"count":0,"items":[],"snapshot_sha256":"258153158e38e3291e3d48162225fcdb2d5a3ed65a07baac614ab91432fd4f57"},"source":{"id":"2606.09001","kind":"arxiv","version":1},"verdict":{"id":null,"model_set":{},"created_at":null,"strongest_claim":"","one_line_summary":"","pipeline_version":null,"weakest_assumption":"","pith_extraction_headline":""},"integrity":{"clean":true,"summary":{"advisory":0,"critical":0,"by_detector":{},"informational":0},"endpoint":"/pith/2606.09001/integrity.json","findings":[],"available":true,"detectors_run":[],"snapshot_sha256":"c28c3603d3b5d939e8dc4c7e95fa8dfce3d595e45f758748cecf8e644a296938"},"references":{"count":0,"sample":[],"resolved_work":0,"snapshot_sha256":"258153158e38e3291e3d48162225fcdb2d5a3ed65a07baac614ab91432fd4f57","internal_anchors":0},"formal_canon":{"evidence_count":0,"snapshot_sha256":"258153158e38e3291e3d48162225fcdb2d5a3ed65a07baac614ab91432fd4f57"},"author_claims":{"count":0,"strong_count":0,"snapshot_sha256":"258153158e38e3291e3d48162225fcdb2d5a3ed65a07baac614ab91432fd4f57"},"builder_version":"pith-number-builder-2026-05-17-v1"},"aliases":[{"alias_kind":"arxiv","alias_value":"2606.09001","created_at":"2026-06-09T02:07:52.819917+00:00"},{"alias_kind":"arxiv_version","alias_value":"2606.09001v1","created_at":"2026-06-09T02:07:52.819917+00:00"},{"alias_kind":"doi","alias_value":"10.48550/arxiv.2606.09001","created_at":"2026-06-09T02:07:52.819917+00:00"},{"alias_kind":"pith_short_12","alias_value":"VMEOFFHDXINC","created_at":"2026-06-09T02:07:52.819917+00:00"},{"alias_kind":"pith_short_16","alias_value":"VMEOFFHDXINCQE2S","created_at":"2026-06-09T02:07:52.819917+00:00"},{"alias_kind":"pith_short_8","alias_value":"VMEOFFHD","created_at":"2026-06-09T02:07:52.819917+00:00"}],"events":[],"event_summary":{},"paper_claims":[],"inbound_citations":{"count":0,"internal_anchor_count":0,"sample":[]},"formal_canon":{"evidence_count":0,"sample":[],"anchors":[]},"links":{"html":"https://pith.science/pith/VMEOFFHDXINCQE2S3O3LSGLZO6","json":"https://pith.science/pith/VMEOFFHDXINCQE2S3O3LSGLZO6.json","graph_json":"https://pith.science/api/pith-number/VMEOFFHDXINCQE2S3O3LSGLZO6/graph.json","events_json":"https://pith.science/api/pith-number/VMEOFFHDXINCQE2S3O3LSGLZO6/events.json","paper":"https://pith.science/paper/VMEOFFHD"},"agent_actions":{"view_html":"https://pith.science/pith/VMEOFFHDXINCQE2S3O3LSGLZO6","download_json":"https://pith.science/pith/VMEOFFHDXINCQE2S3O3LSGLZO6.json","view_paper":"https://pith.science/paper/VMEOFFHD","resolve_alias":"https://pith.science/api/pith-number/resolve?arxiv=2606.09001&json=true","fetch_graph":"https://pith.science/api/pith-number/VMEOFFHDXINCQE2S3O3LSGLZO6/graph.json","fetch_events":"https://pith.science/api/pith-number/VMEOFFHDXINCQE2S3O3LSGLZO6/events.json","actions":{"anchor_timestamp":"https://pith.science/pith/VMEOFFHDXINCQE2S3O3LSGLZO6/action/timestamp_anchor","attest_storage":"https://pith.science/pith/VMEOFFHDXINCQE2S3O3LSGLZO6/action/storage_attestation","attest_author":"https://pith.science/pith/VMEOFFHDXINCQE2S3O3LSGLZO6/action/author_attestation","sign_citation":"https://pith.science/pith/VMEOFFHDXINCQE2S3O3LSGLZO6/action/citation_signature","submit_replication":"https://pith.science/pith/VMEOFFHDXINCQE2S3O3LSGLZO6/action/replication_record"}},"created_at":"2026-06-09T02:07:52.819917+00:00","updated_at":"2026-06-09T02:07:52.819917+00:00"}