{"record_type":"pith_number_record","schema_url":"https://pith.science/schemas/pith-number/v1.json","pith_number":"pith:2026:HBN3YP4ITOW73CEZDD35POPXLH","short_pith_number":"pith:HBN3YP4I","schema_version":"1.0","canonical_sha256":"385bbc3f889badfd889918f7d7b9f759dc2704e0bdb16e5d07c00ff6bfdb4650","source":{"kind":"arxiv","id":"2607.20466","version":1},"attestation_state":"computed","paper":{"title":"JAXBench: Benchmarking Autonomous TPU Kernel Optimization","license":"http://creativecommons.org/licenses/by/4.0/","headline":"","cross_cats":[],"primary_cat":"cs.AI","authors_text":"Amir Yazdanbakhsh, Arya Tschand, Charles Hong, Julian Walker, Nina Cai, Sethu Sankaran, Shangkun Wang, Sundar Dev, Suvinay Subramanian, Vijay Janapa Reddi","submitted_at":"2026-05-19T05:38:39Z","abstract_excerpt":"Rigorous benchmarks have driven progress in autonomous GPU kernel performance optimization by establishing a shared target to hillclimb on, but no equivalent exists for TPUs. We present JAXBench, a TPU-native benchmark suite for AI-generated kernel optimization on Google Cloud TPUs. JAXBench comprises 50 JAX workloads that are both relevant and provide headroom for optimization. We extract 17 production ML operators from architectures in the public MaxText library such as Llama-3.1, DeepSeek-V3, Mixtral, Mamba-2, and AlphaFold2, and translate 33 operators from KernelBench that are validated fo"},"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":"2607.20466","kind":"arxiv","version":1},"metadata":{"license":"http://creativecommons.org/licenses/by/4.0/","primary_cat":"cs.AI","submitted_at":"2026-05-19T05:38:39Z","cross_cats_sorted":[],"title_canon_sha256":"3f838d0ef88c531ca2d15e5d37f2ec62b6adcb34087e8cc37c9ed1da801bd03b","abstract_canon_sha256":"187f83700905480f3337be12fcb4e5d0ced8799eb6a12843f3a7d14dfaf02738"},"schema_version":"1.0"},"receipt":{"kind":"pith_receipt","key_id":"pith-v1-2026-05","algorithm":"ed25519","signed_at":"2026-07-24T00:23:17.169103Z","signature_b64":"SWyhZAZgqh13tblSUAQ24XJ4qEzDOHla+n72b18uH8gIcJsGxJoRgoY6CZO6j9g36zKt7kYuGAwxdwtq74bKDg==","signed_message":"canonical_sha256_bytes","builder_version":"pith-number-builder-2026-05-17-v1","receipt_version":"0.3","canonical_sha256":"385bbc3f889badfd889918f7d7b9f759dc2704e0bdb16e5d07c00ff6bfdb4650","last_reissued_at":"2026-07-24T00:23:17.168230Z","signature_status":"signed_v1","first_computed_at":"2026-07-24T00:23:17.168230Z","public_key_fingerprint":"8d4b5ee74e4693bcd1df2446408b0d54"},"graph_snapshot":{"paper":{"title":"JAXBench: Benchmarking Autonomous TPU Kernel Optimization","license":"http://creativecommons.org/licenses/by/4.0/","headline":"","cross_cats":[],"primary_cat":"cs.AI","authors_text":"Amir Yazdanbakhsh, Arya Tschand, Charles Hong, Julian Walker, Nina Cai, Sethu Sankaran, Shangkun Wang, Sundar Dev, Suvinay Subramanian, Vijay Janapa Reddi","submitted_at":"2026-05-19T05:38:39Z","abstract_excerpt":"Rigorous benchmarks have driven progress in autonomous GPU kernel performance optimization by establishing a shared target to hillclimb on, but no equivalent exists for TPUs. We present JAXBench, a TPU-native benchmark suite for AI-generated kernel optimization on Google Cloud TPUs. JAXBench comprises 50 JAX workloads that are both relevant and provide headroom for optimization. We extract 17 production ML operators from architectures in the public MaxText library such as Llama-3.1, DeepSeek-V3, Mixtral, Mamba-2, and AlphaFold2, and translate 33 operators from KernelBench that are validated fo"},"claims":{"count":0,"items":[],"snapshot_sha256":"258153158e38e3291e3d48162225fcdb2d5a3ed65a07baac614ab91432fd4f57"},"source":{"id":"2607.20466","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/2607.20466/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":"2607.20466","created_at":"2026-07-24T00:23:17.168678+00:00"},{"alias_kind":"arxiv_version","alias_value":"2607.20466v1","created_at":"2026-07-24T00:23:17.168678+00:00"},{"alias_kind":"doi","alias_value":"10.48550/arxiv.2607.20466","created_at":"2026-07-24T00:23:17.168678+00:00"},{"alias_kind":"pith_short_12","alias_value":"HBN3YP4ITOW7","created_at":"2026-07-24T00:23:17.168678+00:00"},{"alias_kind":"pith_short_16","alias_value":"HBN3YP4ITOW73CEZ","created_at":"2026-07-24T00:23:17.168678+00:00"},{"alias_kind":"pith_short_8","alias_value":"HBN3YP4I","created_at":"2026-07-24T00:23:17.168678+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/HBN3YP4ITOW73CEZDD35POPXLH","json":"https://pith.science/pith/HBN3YP4ITOW73CEZDD35POPXLH.json","graph_json":"https://pith.science/api/pith-number/HBN3YP4ITOW73CEZDD35POPXLH/graph.json","events_json":"https://pith.science/api/pith-number/HBN3YP4ITOW73CEZDD35POPXLH/events.json","paper":"https://pith.science/paper/HBN3YP4I"},"agent_actions":{"view_html":"https://pith.science/pith/HBN3YP4ITOW73CEZDD35POPXLH","download_json":"https://pith.science/pith/HBN3YP4ITOW73CEZDD35POPXLH.json","view_paper":"https://pith.science/paper/HBN3YP4I","resolve_alias":"https://pith.science/api/pith-number/resolve?arxiv=2607.20466&json=true","fetch_graph":"https://pith.science/api/pith-number/HBN3YP4ITOW73CEZDD35POPXLH/graph.json","fetch_events":"https://pith.science/api/pith-number/HBN3YP4ITOW73CEZDD35POPXLH/events.json","actions":{"anchor_timestamp":"https://pith.science/pith/HBN3YP4ITOW73CEZDD35POPXLH/action/timestamp_anchor","attest_storage":"https://pith.science/pith/HBN3YP4ITOW73CEZDD35POPXLH/action/storage_attestation","attest_author":"https://pith.science/pith/HBN3YP4ITOW73CEZDD35POPXLH/action/author_attestation","sign_citation":"https://pith.science/pith/HBN3YP4ITOW73CEZDD35POPXLH/action/citation_signature","submit_replication":"https://pith.science/pith/HBN3YP4ITOW73CEZDD35POPXLH/action/replication_record"}},"created_at":"2026-07-24T00:23:17.168678+00:00","updated_at":"2026-07-24T00:23:17.168678+00:00"}