{"bundle_type":"pith_open_graph_bundle","bundle_version":"1.0","pith_number":"pith:2019:52FPSCX54XKSAO7XWVIESKS6JR","short_pith_number":"pith:52FPSCX5","canonical_record":{"source":{"id":"1912.11554","kind":"arxiv","version":1},"metadata":{"license":"http://arxiv.org/licenses/nonexclusive-distrib/1.0/","primary_cat":"stat.ML","submitted_at":"2019-12-24T22:09:36Z","cross_cats_sorted":["cs.AI","cs.LG","cs.PL"],"title_canon_sha256":"4cada3a002c300a0256c850b4de223b1aecb15768cfd628ce5059fb220f3bb79","abstract_canon_sha256":"23b4291795aec701f8a107f83cc5925ae9b17f75d4488a1480b7e616c2a31232"},"schema_version":"1.0"},"canonical_sha256":"ee8af90afde5d5203bf7b550492a5e4c46bd268faf40cd2a835c37b32cb09d3c","source":{"kind":"arxiv","id":"1912.11554","version":1},"source_aliases":[{"alias_kind":"arxiv","alias_value":"1912.11554","created_at":"2026-05-17T23:38:50Z"},{"alias_kind":"arxiv_version","alias_value":"1912.11554v1","created_at":"2026-05-17T23:38:50Z"},{"alias_kind":"doi","alias_value":"10.48550/arxiv.1912.11554","created_at":"2026-05-17T23:38:50Z"},{"alias_kind":"pith_short_12","alias_value":"52FPSCX54XKS","created_at":"2026-05-18T12:33:10Z"},{"alias_kind":"pith_short_16","alias_value":"52FPSCX54XKSAO7X","created_at":"2026-05-18T12:33:10Z"},{"alias_kind":"pith_short_8","alias_value":"52FPSCX5","created_at":"2026-05-18T12:33:10Z"}],"events":[{"event_type":"record_created","subject_pith_number":"pith:2019:52FPSCX54XKSAO7XWVIESKS6JR","target":"record","payload":{"canonical_record":{"source":{"id":"1912.11554","kind":"arxiv","version":1},"metadata":{"license":"http://arxiv.org/licenses/nonexclusive-distrib/1.0/","primary_cat":"stat.ML","submitted_at":"2019-12-24T22:09:36Z","cross_cats_sorted":["cs.AI","cs.LG","cs.PL"],"title_canon_sha256":"4cada3a002c300a0256c850b4de223b1aecb15768cfd628ce5059fb220f3bb79","abstract_canon_sha256":"23b4291795aec701f8a107f83cc5925ae9b17f75d4488a1480b7e616c2a31232"},"schema_version":"1.0"},"canonical_sha256":"ee8af90afde5d5203bf7b550492a5e4c46bd268faf40cd2a835c37b32cb09d3c","receipt":{"kind":"pith_receipt","key_id":"pith-v1-2026-05","algorithm":"ed25519","signed_at":"2026-05-17T23:38:50.993110Z","signature_b64":"Sv0fpw+I8FxFZaKLXISi68PVN6JElAXArY5T1eWTLt24CwTJRmAuYnvu9zIXpdMwaDV6+o6KLV/azWL61ovnBQ==","signed_message":"canonical_sha256_bytes","builder_version":"pith-number-builder-2026-05-17-v1","receipt_version":"0.3","canonical_sha256":"ee8af90afde5d5203bf7b550492a5e4c46bd268faf40cd2a835c37b32cb09d3c","last_reissued_at":"2026-05-17T23:38:50.992672Z","signature_status":"signed_v1","first_computed_at":"2026-05-17T23:38:50.992672Z","public_key_fingerprint":"8d4b5ee74e4693bcd1df2446408b0d54"},"source_kind":"arxiv","source_id":"1912.11554","source_version":1,"attestation_state":"computed"},"signer":{"signer_id":"pith.science","signer_type":"pith_registry","key_id":"pith-v1-2026-05","public_key_fingerprint":"8d4b5ee74e4693bcd1df2446408b0d54"},"created_at":"2026-05-17T23:38:50Z","supersedes":[],"prev_event":null,"signature":{"signature_status":"signed_v1","algorithm":"ed25519","key_id":"pith-v1-2026-05","public_key_fingerprint":"8d4b5ee74e4693bcd1df2446408b0d54","signature_b64":"84UQYfA3OSyBRDD/Ttf44S3J23z44rjj+7Kke8WzfxUK50FgCEjdZboPhuwj5AvU10oTzEoXf4U2u8O/dUKCCQ==","signed_message":"open_graph_event_sha256_bytes","signed_at":"2026-05-25T22:21:17.142423Z"},"content_sha256":"1bb656228fe26d996fc7412e7656e9300b4f93d42b65e3b6b0a7f5989f13177b","schema_version":"1.0","event_id":"sha256:1bb656228fe26d996fc7412e7656e9300b4f93d42b65e3b6b0a7f5989f13177b"},{"event_type":"graph_snapshot","subject_pith_number":"pith:2019:52FPSCX54XKSAO7XWVIESKS6JR","target":"graph","payload":{"graph_snapshot":{"paper":{"title":"Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro","license":"http://arxiv.org/licenses/nonexclusive-distrib/1.0/","headline":"NumPyro composes Pyro effect handlers with JAX to deliver a fully JIT-compiled iterative NUTS sampler.","cross_cats":["cs.AI","cs.LG","cs.PL"],"primary_cat":"stat.ML","authors_text":"Du Phan, Martin Jankowiak, Neeraj Pradhan","submitted_at":"2019-12-24T22:09:36Z","abstract_excerpt":"NumPyro is a lightweight library that provides an alternate NumPy backend to the Pyro probabilistic programming language with the same modeling interface, language primitives and effect handling abstractions. Effect handlers allow Pyro's modeling API to be extended to NumPyro despite its being built atop a fundamentally different JAX-based functional backend. In this work, we demonstrate the power of composing Pyro's effect handlers with the program transformations that enable hardware acceleration, automatic differentiation, and vectorization in JAX. In particular, NumPyro provides an iterati"},"claims":{"count":4,"items":[{"kind":"strongest_claim","text":"NumPyro provides an iterative formulation of the No-U-Turn Sampler (NUTS) that can be end-to-end JIT compiled, yielding an implementation that is much faster than existing alternatives in both the small and large dataset regimes.","source":"verdict.strongest_claim","status":"machine_extracted","claim_id":"C1","attestation":"unclaimed"},{"kind":"weakest_assumption","text":"That Pyro's effect handlers compose cleanly with JAX's functional transformations without introducing correctness issues or losing expressiveness in the modeling API.","source":"verdict.weakest_assumption","status":"machine_extracted","claim_id":"C2","attestation":"unclaimed"},{"kind":"one_line_summary","text":"NumPyro delivers a JIT-compilable iterative NUTS sampler by composing Pyro effect handlers with JAX transformations, achieving faster performance than prior implementations.","source":"verdict.one_line_summary","status":"machine_extracted","claim_id":"C3","attestation":"unclaimed"},{"kind":"headline","text":"NumPyro composes Pyro effect handlers with JAX to deliver a fully JIT-compiled iterative NUTS sampler.","source":"verdict.pith_extraction.headline","status":"machine_extracted","claim_id":"C4","attestation":"unclaimed"}],"snapshot_sha256":"10f1c0fd0a5e87955ee0e1aae058fed1c814acb0fd873065a02895e3496f2689"},"source":{"id":"1912.11554","kind":"arxiv","version":1},"verdict":{"id":"ec506122-a049-4c3b-ab7e-3c0f7dd3568d","model_set":{"reader":"grok-4.3"},"created_at":"2026-05-15T15:58:48.146930Z","strongest_claim":"NumPyro provides an iterative formulation of the No-U-Turn Sampler (NUTS) that can be end-to-end JIT compiled, yielding an implementation that is much faster than existing alternatives in both the small and large dataset regimes.","one_line_summary":"NumPyro delivers a JIT-compilable iterative NUTS sampler by composing Pyro effect handlers with JAX transformations, achieving faster performance than prior implementations.","pipeline_version":"pith-pipeline@v0.9.0","weakest_assumption":"That Pyro's effect handlers compose cleanly with JAX's functional transformations without introducing correctness issues or losing expressiveness in the modeling API.","pith_extraction_headline":"NumPyro composes Pyro effect handlers with JAX to deliver a fully JIT-compiled iterative NUTS sampler."},"references":{"count":25,"sample":[{"doi":"","year":1912,"title":"Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro","work_id":"a2a03fcf-38c5-4ebe-b5a4-4bc41761dca5","ref_index":1,"cited_arxiv_id":"1912.11554","is_internal_anchor":true},{"doi":"","year":2018,"title":"2018 , copyright =","work_id":"e0d9664f-f519-4c55-b035-0f5e384b0835","ref_index":2,"cited_arxiv_id":"","is_internal_anchor":false},{"doi":"","year":2019,"title":"Chen, Martin Jankowiak, Fritz Obermeyer, Neeraj Pradhan, Theofanis Karaletsos, Rohit Singh, Paul Szerlip, Paul Horsfall, and Noah D","work_id":"f70b8d95-49b2-402e-a34c-887bb1b81626","ref_index":3,"cited_arxiv_id":"","is_internal_anchor":false},{"doi":"","year":2017,"title":"Learning disentangled representations with semi-supervised deep generative models","work_id":"d809c362-1be9-4817-bbc6-2a5c9cdc6bfe","ref_index":4,"cited_arxiv_id":"","is_internal_anchor":false},{"doi":"","year":2017,"title":"Automatic differentiation in pytorch","work_id":"5a5818ac-37e1-444f-8802-6256d3fb52b9","ref_index":5,"cited_arxiv_id":"","is_internal_anchor":false}],"resolved_work":25,"snapshot_sha256":"2b88815c921d3cad2a3bf203a755ced74764ff24690900947d68ec2b6b187b3a","internal_anchors":3},"formal_canon":{"evidence_count":2,"snapshot_sha256":"7275956bfa88188bfc6605fa18cb416fe8164b8e578a83f3d3ee802222110e4f"},"author_claims":{"count":0,"strong_count":0,"snapshot_sha256":"258153158e38e3291e3d48162225fcdb2d5a3ed65a07baac614ab91432fd4f57"},"builder_version":"pith-number-builder-2026-05-17-v1"},"verdict_id":"ec506122-a049-4c3b-ab7e-3c0f7dd3568d"},"signer":{"signer_id":"pith.science","signer_type":"pith_registry","key_id":"pith-v1-2026-05","public_key_fingerprint":"8d4b5ee74e4693bcd1df2446408b0d54"},"created_at":"2026-05-17T23:38:50Z","supersedes":[],"prev_event":null,"signature":{"signature_status":"signed_v1","algorithm":"ed25519","key_id":"pith-v1-2026-05","public_key_fingerprint":"8d4b5ee74e4693bcd1df2446408b0d54","signature_b64":"uZgqNzE3l2KQ5X8R2fdsaiehSkyMyE6rCj/nfrKNXnuUjpvD+bxlwcxjekWclFNOoAjE904Qrs1MgGKiB38uAA==","signed_message":"open_graph_event_sha256_bytes","signed_at":"2026-05-25T22:21:17.143228Z"},"content_sha256":"7230bf9ea78d2c406f4aa793b6a6f616403566a1775c03fdc69621b6466d51c1","schema_version":"1.0","event_id":"sha256:7230bf9ea78d2c406f4aa793b6a6f616403566a1775c03fdc69621b6466d51c1"}],"timestamp_proofs":[],"mirror_hints":[{"mirror_type":"https","name":"Pith Resolver","base_url":"https://pith.science","bundle_url":"https://pith.science/pith/52FPSCX54XKSAO7XWVIESKS6JR/bundle.json","state_url":"https://pith.science/pith/52FPSCX54XKSAO7XWVIESKS6JR/state.json","well_known_bundle_url":"https://pith.science/.well-known/pith/52FPSCX54XKSAO7XWVIESKS6JR/bundle.json","status":"primary"}],"public_keys":[{"key_id":"pith-v1-2026-05","algorithm":"ed25519","format":"raw","public_key_b64":"stVStoiQhXFxp4s2pdzPNoqVNBMojDU/fJ2db5S3CbM=","public_key_hex":"b2d552b68890857171a78b36a5dccf368a953413288c353f7c9d9d6f94b709b3","fingerprint_sha256_b32_first128bits":"RVFV5Z2OI2J3ZUO7ERDEBCYNKS","fingerprint_sha256_hex":"8d4b5ee74e4693bcd1df2446408b0d54","rotates_at":null,"url":"https://pith.science/pith-signing-key.json","notes":"Pith uses this Ed25519 key to sign canonical record SHA-256 digests. Verify with: ed25519_verify(public_key, message=canonical_sha256_bytes, signature=base64decode(signature_b64))."}],"merge_version":"pith-open-graph-merge-v1","built_at":"2026-05-25T22:21:17Z","links":{"resolver":"https://pith.science/pith/52FPSCX54XKSAO7XWVIESKS6JR","bundle":"https://pith.science/pith/52FPSCX54XKSAO7XWVIESKS6JR/bundle.json","state":"https://pith.science/pith/52FPSCX54XKSAO7XWVIESKS6JR/state.json","well_known_bundle":"https://pith.science/.well-known/pith/52FPSCX54XKSAO7XWVIESKS6JR/bundle.json"},"state":{"state_type":"pith_open_graph_state","state_version":"1.0","pith_number":"pith:2019:52FPSCX54XKSAO7XWVIESKS6JR","merge_version":"pith-open-graph-merge-v1","event_count":2,"valid_event_count":2,"invalid_event_count":0,"equivocation_count":0,"current":{"canonical_record":{"metadata":{"abstract_canon_sha256":"23b4291795aec701f8a107f83cc5925ae9b17f75d4488a1480b7e616c2a31232","cross_cats_sorted":["cs.AI","cs.LG","cs.PL"],"license":"http://arxiv.org/licenses/nonexclusive-distrib/1.0/","primary_cat":"stat.ML","submitted_at":"2019-12-24T22:09:36Z","title_canon_sha256":"4cada3a002c300a0256c850b4de223b1aecb15768cfd628ce5059fb220f3bb79"},"schema_version":"1.0","source":{"id":"1912.11554","kind":"arxiv","version":1}},"source_aliases":[{"alias_kind":"arxiv","alias_value":"1912.11554","created_at":"2026-05-17T23:38:50Z"},{"alias_kind":"arxiv_version","alias_value":"1912.11554v1","created_at":"2026-05-17T23:38:50Z"},{"alias_kind":"doi","alias_value":"10.48550/arxiv.1912.11554","created_at":"2026-05-17T23:38:50Z"},{"alias_kind":"pith_short_12","alias_value":"52FPSCX54XKS","created_at":"2026-05-18T12:33:10Z"},{"alias_kind":"pith_short_16","alias_value":"52FPSCX54XKSAO7X","created_at":"2026-05-18T12:33:10Z"},{"alias_kind":"pith_short_8","alias_value":"52FPSCX5","created_at":"2026-05-18T12:33:10Z"}],"graph_snapshots":[{"event_id":"sha256:7230bf9ea78d2c406f4aa793b6a6f616403566a1775c03fdc69621b6466d51c1","target":"graph","created_at":"2026-05-17T23:38:50Z","signer":{"key_id":"pith-v1-2026-05","public_key_fingerprint":"8d4b5ee74e4693bcd1df2446408b0d54","signer_id":"pith.science","signer_type":"pith_registry"},"payload":{"graph_snapshot":{"author_claims":{"count":0,"snapshot_sha256":"258153158e38e3291e3d48162225fcdb2d5a3ed65a07baac614ab91432fd4f57","strong_count":0},"builder_version":"pith-number-builder-2026-05-17-v1","claims":{"count":4,"items":[{"attestation":"unclaimed","claim_id":"C1","kind":"strongest_claim","source":"verdict.strongest_claim","status":"machine_extracted","text":"NumPyro provides an iterative formulation of the No-U-Turn Sampler (NUTS) that can be end-to-end JIT compiled, yielding an implementation that is much faster than existing alternatives in both the small and large dataset regimes."},{"attestation":"unclaimed","claim_id":"C2","kind":"weakest_assumption","source":"verdict.weakest_assumption","status":"machine_extracted","text":"That Pyro's effect handlers compose cleanly with JAX's functional transformations without introducing correctness issues or losing expressiveness in the modeling API."},{"attestation":"unclaimed","claim_id":"C3","kind":"one_line_summary","source":"verdict.one_line_summary","status":"machine_extracted","text":"NumPyro delivers a JIT-compilable iterative NUTS sampler by composing Pyro effect handlers with JAX transformations, achieving faster performance than prior implementations."},{"attestation":"unclaimed","claim_id":"C4","kind":"headline","source":"verdict.pith_extraction.headline","status":"machine_extracted","text":"NumPyro composes Pyro effect handlers with JAX to deliver a fully JIT-compiled iterative NUTS sampler."}],"snapshot_sha256":"10f1c0fd0a5e87955ee0e1aae058fed1c814acb0fd873065a02895e3496f2689"},"formal_canon":{"evidence_count":2,"snapshot_sha256":"7275956bfa88188bfc6605fa18cb416fe8164b8e578a83f3d3ee802222110e4f"},"paper":{"abstract_excerpt":"NumPyro is a lightweight library that provides an alternate NumPy backend to the Pyro probabilistic programming language with the same modeling interface, language primitives and effect handling abstractions. Effect handlers allow Pyro's modeling API to be extended to NumPyro despite its being built atop a fundamentally different JAX-based functional backend. In this work, we demonstrate the power of composing Pyro's effect handlers with the program transformations that enable hardware acceleration, automatic differentiation, and vectorization in JAX. In particular, NumPyro provides an iterati","authors_text":"Du Phan, Martin Jankowiak, Neeraj Pradhan","cross_cats":["cs.AI","cs.LG","cs.PL"],"headline":"NumPyro composes Pyro effect handlers with JAX to deliver a fully JIT-compiled iterative NUTS sampler.","license":"http://arxiv.org/licenses/nonexclusive-distrib/1.0/","primary_cat":"stat.ML","submitted_at":"2019-12-24T22:09:36Z","title":"Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro"},"references":{"count":25,"internal_anchors":3,"resolved_work":25,"sample":[{"cited_arxiv_id":"1912.11554","doi":"","is_internal_anchor":true,"ref_index":1,"title":"Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyro","work_id":"a2a03fcf-38c5-4ebe-b5a4-4bc41761dca5","year":1912},{"cited_arxiv_id":"","doi":"","is_internal_anchor":false,"ref_index":2,"title":"2018 , copyright =","work_id":"e0d9664f-f519-4c55-b035-0f5e384b0835","year":2018},{"cited_arxiv_id":"","doi":"","is_internal_anchor":false,"ref_index":3,"title":"Chen, Martin Jankowiak, Fritz Obermeyer, Neeraj Pradhan, Theofanis Karaletsos, Rohit Singh, Paul Szerlip, Paul Horsfall, and Noah D","work_id":"f70b8d95-49b2-402e-a34c-887bb1b81626","year":2019},{"cited_arxiv_id":"","doi":"","is_internal_anchor":false,"ref_index":4,"title":"Learning disentangled representations with semi-supervised deep generative models","work_id":"d809c362-1be9-4817-bbc6-2a5c9cdc6bfe","year":2017},{"cited_arxiv_id":"","doi":"","is_internal_anchor":false,"ref_index":5,"title":"Automatic differentiation in pytorch","work_id":"5a5818ac-37e1-444f-8802-6256d3fb52b9","year":2017}],"snapshot_sha256":"2b88815c921d3cad2a3bf203a755ced74764ff24690900947d68ec2b6b187b3a"},"source":{"id":"1912.11554","kind":"arxiv","version":1},"verdict":{"created_at":"2026-05-15T15:58:48.146930Z","id":"ec506122-a049-4c3b-ab7e-3c0f7dd3568d","model_set":{"reader":"grok-4.3"},"one_line_summary":"NumPyro delivers a JIT-compilable iterative NUTS sampler by composing Pyro effect handlers with JAX transformations, achieving faster performance than prior implementations.","pipeline_version":"pith-pipeline@v0.9.0","pith_extraction_headline":"NumPyro composes Pyro effect handlers with JAX to deliver a fully JIT-compiled iterative NUTS sampler.","strongest_claim":"NumPyro provides an iterative formulation of the No-U-Turn Sampler (NUTS) that can be end-to-end JIT compiled, yielding an implementation that is much faster than existing alternatives in both the small and large dataset regimes.","weakest_assumption":"That Pyro's effect handlers compose cleanly with JAX's functional transformations without introducing correctness issues or losing expressiveness in the modeling API."}},"verdict_id":"ec506122-a049-4c3b-ab7e-3c0f7dd3568d"}}],"author_attestations":[],"timestamp_anchors":[],"storage_attestations":[],"citation_signatures":[],"replication_records":[],"corrections":[],"mirror_hints":[],"record_created":{"event_id":"sha256:1bb656228fe26d996fc7412e7656e9300b4f93d42b65e3b6b0a7f5989f13177b","target":"record","created_at":"2026-05-17T23:38:50Z","signer":{"key_id":"pith-v1-2026-05","public_key_fingerprint":"8d4b5ee74e4693bcd1df2446408b0d54","signer_id":"pith.science","signer_type":"pith_registry"},"payload":{"attestation_state":"computed","canonical_record":{"metadata":{"abstract_canon_sha256":"23b4291795aec701f8a107f83cc5925ae9b17f75d4488a1480b7e616c2a31232","cross_cats_sorted":["cs.AI","cs.LG","cs.PL"],"license":"http://arxiv.org/licenses/nonexclusive-distrib/1.0/","primary_cat":"stat.ML","submitted_at":"2019-12-24T22:09:36Z","title_canon_sha256":"4cada3a002c300a0256c850b4de223b1aecb15768cfd628ce5059fb220f3bb79"},"schema_version":"1.0","source":{"id":"1912.11554","kind":"arxiv","version":1}},"canonical_sha256":"ee8af90afde5d5203bf7b550492a5e4c46bd268faf40cd2a835c37b32cb09d3c","receipt":{"algorithm":"ed25519","builder_version":"pith-number-builder-2026-05-17-v1","canonical_sha256":"ee8af90afde5d5203bf7b550492a5e4c46bd268faf40cd2a835c37b32cb09d3c","first_computed_at":"2026-05-17T23:38:50.992672Z","key_id":"pith-v1-2026-05","kind":"pith_receipt","last_reissued_at":"2026-05-17T23:38:50.992672Z","public_key_fingerprint":"8d4b5ee74e4693bcd1df2446408b0d54","receipt_version":"0.3","signature_b64":"Sv0fpw+I8FxFZaKLXISi68PVN6JElAXArY5T1eWTLt24CwTJRmAuYnvu9zIXpdMwaDV6+o6KLV/azWL61ovnBQ==","signature_status":"signed_v1","signed_at":"2026-05-17T23:38:50.993110Z","signed_message":"canonical_sha256_bytes"},"source_id":"1912.11554","source_kind":"arxiv","source_version":1}}},"equivocations":[],"invalid_events":[],"applied_event_ids":["sha256:1bb656228fe26d996fc7412e7656e9300b4f93d42b65e3b6b0a7f5989f13177b","sha256:7230bf9ea78d2c406f4aa793b6a6f616403566a1775c03fdc69621b6466d51c1"],"state_sha256":"75d9e3e4ac4a35927c010eddd3f321d1ea4b3d1f37837c386d38354a04daf26d"},"bundle_signature":{"signature_status":"signed_v1","algorithm":"ed25519","key_id":"pith-v1-2026-05","public_key_fingerprint":"8d4b5ee74e4693bcd1df2446408b0d54","signature_b64":"o7A1OkwFCV/7gEkVbeASQyHQOtr9J7+f57IozIlRuHob58jezHeTMDXWqFz6d9JWkL8KEr1nN9Y7l+FoewmiAg==","signed_message":"bundle_sha256_bytes","signed_at":"2026-05-25T22:21:17.149198Z","bundle_sha256":"0bf94d78e1cc02688d095b544fd6f94f6b54d12f3e4f1817ab0e712fcc64f5ac"}}