{"id":"https://openalex.org/W7161823003","doi":"https://doi.org/10.48550/arxiv.2605.19269","title":"CODA: Rewriting Transformer Blocks as GEMM-Epilogue Programs","display_name":"CODA: Rewriting Transformer Blocks as GEMM-Epilogue Programs","publication_year":2026,"publication_date":"2026-05-19","ids":{"openalex":"https://openalex.org/W7161823003","doi":"https://doi.org/10.48550/arxiv.2605.19269"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.19269","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.19269","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2605.19269","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5136594505","display_name":"Han Guo","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Guo, Han","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5135957135","display_name":"Jack Zhang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Jack","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136604517","display_name":"Arjun Menon","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Menon, Arjun","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136509258","display_name":"Driss Guessous","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Guessous, Driss","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5086615721","display_name":"Vijay Thakkar","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Thakkar, Vijay","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136536438","display_name":"Yoon Kim","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kim, Yoon","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5136536267","display_name":"Tri Dao","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Dao, Tri","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":0,"corresponding_author_ids":[],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":false,"cited_by_count":0,"citation_normalized_percentile":null,"cited_by_percentile_year":null,"biblio":{"volume":null,"issue":null,"first_page":null,"last_page":null},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.9603999853134155,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T10054","display_name":"Parallel Computing and Optimization Techniques","score":0.9603999853134155,"subfield":{"id":"https://openalex.org/subfields/1708","display_name":"Hardware and Architecture"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T12808","display_name":"Ferroelectric and Negative Capacitance Devices","score":0.00419999985024333,"subfield":{"id":"https://openalex.org/subfields/2208","display_name":"Electrical and Electronic Engineering"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10126","display_name":"Logic, programming, and type systems","score":0.004000000189989805,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/transformer","display_name":"Transformer","score":0.5888000130653381},{"id":"https://openalex.org/keywords/computation","display_name":"Computation","score":0.5540000200271606},{"id":"https://openalex.org/keywords/rewriting","display_name":"Rewriting","score":0.5085999965667725},{"id":"https://openalex.org/keywords/embedding","display_name":"Embedding","score":0.4738999903202057},{"id":"https://openalex.org/keywords/bottleneck","display_name":"Bottleneck","score":0.35109999775886536},{"id":"https://openalex.org/keywords/abstraction","display_name":"Abstraction","score":0.3197999894618988}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6825000047683716},{"id":"https://openalex.org/C66322947","wikidata":"https://www.wikidata.org/wiki/Q11658","display_name":"Transformer","level":3,"score":0.5888000130653381},{"id":"https://openalex.org/C45374587","wikidata":"https://www.wikidata.org/wiki/Q12525525","display_name":"Computation","level":2,"score":0.5540000200271606},{"id":"https://openalex.org/C154690210","wikidata":"https://www.wikidata.org/wiki/Q1668499","display_name":"Rewriting","level":2,"score":0.5085999965667725},{"id":"https://openalex.org/C199360897","wikidata":"https://www.wikidata.org/wiki/Q9143","display_name":"Programming language","level":1,"score":0.499099999666214},{"id":"https://openalex.org/C41608201","wikidata":"https://www.wikidata.org/wiki/Q980509","display_name":"Embedding","level":2,"score":0.4738999903202057},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.3817000091075897},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.36079999804496765},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.35120001435279846},{"id":"https://openalex.org/C2780513914","wikidata":"https://www.wikidata.org/wiki/Q18210350","display_name":"Bottleneck","level":2,"score":0.35109999775886536},{"id":"https://openalex.org/C124304363","wikidata":"https://www.wikidata.org/wiki/Q673661","display_name":"Abstraction","level":2,"score":0.3197999894618988},{"id":"https://openalex.org/C74193536","wikidata":"https://www.wikidata.org/wiki/Q574844","display_name":"Kernel (algebra)","level":2,"score":0.3122999966144562},{"id":"https://openalex.org/C114466953","wikidata":"https://www.wikidata.org/wiki/Q6034165","display_name":"Initialization","level":2,"score":0.2842999994754791},{"id":"https://openalex.org/C193702766","wikidata":"https://www.wikidata.org/wiki/Q1414548","display_name":"Concurrency","level":2,"score":0.2840000092983246},{"id":"https://openalex.org/C184898388","wikidata":"https://www.wikidata.org/wiki/Q1435712","display_name":"Pairwise comparison","level":2,"score":0.2696000039577484},{"id":"https://openalex.org/C2776937632","wikidata":"https://www.wikidata.org/wiki/Q4117718","display_name":"Program synthesis","level":2,"score":0.25679999589920044},{"id":"https://openalex.org/C191399111","wikidata":"https://www.wikidata.org/wiki/Q64861","display_name":"Transitive relation","level":2,"score":0.25029999017715454}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.19269","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.19269","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2605.19269","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.19269","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":"2331-8422","issn":["2331-8422"],"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":null,"license_id":null,"version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"sustainable_development_goals":[{"score":0.5556173324584961,"id":"https://metadata.un.org/sdg/8","display_name":"Decent work and economic growth"}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Transformer":[0,77,155,159],"training":[1,53],"systems":[2],"are":[3],"built":[4],"around":[5],"dense":[6],"linear":[7],"algebra,":[8],"yet":[9],"a":[10,58,91,112,153,175],"nontrivial":[11],"fraction":[12],"of":[13,115,133,152],"end-to-end":[14],"time":[15],"is":[16,70,100],"spent":[17],"on":[18,72,96],"surrounding":[19],"memory-bound":[20],"operators.":[21],"Normalization,":[22],"activations,":[23],"residual":[24],"updates,":[25],"reductions,":[26,121],"and":[27,110,124,149,163],"related":[28],"computations":[29,65],"repeatedly":[30],"move":[31],"large":[32],"intermediate":[33],"tensors":[34],"through":[35],"global":[36],"memory":[37],"while":[38,90,136],"performing":[39],"little":[40],"arithmetic,":[41],"making":[42],"data":[43],"movement":[44],"an":[45],"increasingly":[46],"important":[47],"bottleneck":[48],"in":[49,146],"otherwise":[50],"highly":[51],"optimized":[52],"stacks.":[54],"We":[55],"introduce":[56],"CODA,":[57],"GPU":[59],"kernel":[60],"abstraction":[61,105],"that":[62,75,171],"expresses":[63],"these":[64],"as":[66,80],"GEMM-plus-epilogue":[67,172],"programs.":[68],"CODA":[69,165],"based":[71],"the":[73,107,130,147],"observation":[74],"many":[76],"operators":[78],"exposed":[79],"separate":[81],"framework":[82],"kernels":[83,166],"can":[84],"be":[85],"algebraically":[86],"reparameterized":[87],"to":[88,102,140],"execute":[89],"GEMM":[92,108],"output":[93],"tile":[94],"remains":[95],"chip,":[97],"before":[98],"it":[99],"written":[101],"memory.":[103],"The":[104],"fixes":[106],"mainloop":[109],"exposes":[111],"small":[113],"set":[114],"composable":[116],"epilogue":[117],"primitives":[118],"for":[119],"scaling,":[120],"pairwise":[122],"transformations,":[123],"accumulation.":[125],"This":[126],"constrained":[127],"interface":[128],"preserves":[129],"performance":[131],"structure":[132],"expert-written":[134],"GEMMs":[135],"remaining":[137],"expressive":[138],"enough":[139],"cover":[141],"nearly":[142],"all":[143],"non-attention":[144],"computation":[145],"forward":[148],"backward":[150],"pass":[151],"standard":[154],"block.":[156],"Across":[157],"representative":[158],"workloads,":[160],"both":[161],"human-":[162],"LLM-authored":[164],"achieve":[167],"high":[168],"performance,":[169],"suggesting":[170],"programming":[173],"offers":[174],"practical":[176],"path":[177],"toward":[178],"combining":[179],"framework-level":[180],"productivity":[181],"with":[182],"hardware-level":[183],"efficiency.":[184]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-05-21T00:00:00"}