{"id":"https://openalex.org/W7147119658","doi":"https://doi.org/10.48550/arxiv.2603.27296","title":"A Multi-agent AI System for Deep Learning Model Migration from TensorFlow to JAX","display_name":"A Multi-agent AI System for Deep Learning Model Migration from TensorFlow to JAX","publication_year":2026,"publication_date":"2026-03-28","ids":{"openalex":"https://openalex.org/W7147119658","doi":"https://doi.org/10.48550/arxiv.2603.27296"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2603.27296","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.27296","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"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":"cc-by","license_id":"https://openalex.org/licenses/cc-by","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.2603.27296","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5086488734","display_name":"Stoyan C. Nikolov","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Nikolov, Stoyan","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5132579842","display_name":"Bernhard Konrad","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Konrad, Bernhard","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5132671423","display_name":"Moritz Gronbach","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Gronbach, Moritz","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5113932716","display_name":"Niket Kumar","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kumar, Niket","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5126198818","display_name":"Ann Yan","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yan, Ann","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5067526759","display_name":"Varun Kumar Singh","orcid":"https://orcid.org/0000-0001-8735-3834"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Singh, Varun","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5100847271","display_name":"Yaning Liang","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Liang, Yaning","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5011491987","display_name":"Parthasarathy Ranganathan","orcid":"https://orcid.org/0000-0002-9751-5902"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ranganathan, Parthasarathy","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/T10260","display_name":"Software Engineering Research","score":0.4659999907016754,"subfield":{"id":"https://openalex.org/subfields/1710","display_name":"Information Systems"},"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/T10260","display_name":"Software Engineering Research","score":0.4659999907016754,"subfield":{"id":"https://openalex.org/subfields/1710","display_name":"Information Systems"},"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/T12535","display_name":"Machine Learning and Data Classification","score":0.06520000100135803,"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"}},{"id":"https://openalex.org/T11689","display_name":"Adversarial Robustness in Machine Learning","score":0.05810000002384186,"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/deep-learning","display_name":"Deep learning","score":0.7678999900817871},{"id":"https://openalex.org/keywords/code","display_name":"Code (set theory)","score":0.6245999932289124},{"id":"https://openalex.org/keywords/planner","display_name":"Planner","score":0.54830002784729},{"id":"https://openalex.org/keywords/source-code","display_name":"Source code","score":0.4228000044822693},{"id":"https://openalex.org/keywords/dependency","display_name":"Dependency (UML)","score":0.3849000036716461},{"id":"https://openalex.org/keywords/quality","display_name":"Quality (philosophy)","score":0.3817000091075897}],"concepts":[{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.7720999717712402},{"id":"https://openalex.org/C108583219","wikidata":"https://www.wikidata.org/wiki/Q197536","display_name":"Deep learning","level":2,"score":0.7678999900817871},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7437000274658203},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.6245999932289124},{"id":"https://openalex.org/C2776999362","wikidata":"https://www.wikidata.org/wiki/Q2349274","display_name":"Planner","level":2,"score":0.54830002784729},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.46209999918937683},{"id":"https://openalex.org/C43126263","wikidata":"https://www.wikidata.org/wiki/Q128751","display_name":"Source code","level":2,"score":0.4228000044822693},{"id":"https://openalex.org/C19768560","wikidata":"https://www.wikidata.org/wiki/Q320727","display_name":"Dependency (UML)","level":2,"score":0.3849000036716461},{"id":"https://openalex.org/C2779530757","wikidata":"https://www.wikidata.org/wiki/Q1207505","display_name":"Quality (philosophy)","level":2,"score":0.3817000091075897},{"id":"https://openalex.org/C67186912","wikidata":"https://www.wikidata.org/wiki/Q367664","display_name":"Data modeling","level":2,"score":0.33149999380111694},{"id":"https://openalex.org/C26517878","wikidata":"https://www.wikidata.org/wiki/Q228039","display_name":"Key (lock)","level":2,"score":0.2833999991416931},{"id":"https://openalex.org/C204241405","wikidata":"https://www.wikidata.org/wiki/Q461499","display_name":"Transformation (genetics)","level":3,"score":0.2759999930858612},{"id":"https://openalex.org/C98045186","wikidata":"https://www.wikidata.org/wiki/Q205663","display_name":"Process (computing)","level":2,"score":0.2678999900817871},{"id":"https://openalex.org/C48044578","wikidata":"https://www.wikidata.org/wiki/Q727490","display_name":"Scalability","level":2,"score":0.2587999999523163},{"id":"https://openalex.org/C25343380","wikidata":"https://www.wikidata.org/wiki/Q277521","display_name":"Relation (database)","level":2,"score":0.25519999861717224}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2603.27296","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.27296","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"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":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2603.27296","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.27296","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"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":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"sustainable_development_goals":[{"display_name":"Reduced inequalities","id":"https://metadata.un.org/sdg/10","score":0.6362254619598389}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"The":[0],"rapid":[1],"development":[2,139,200],"of":[3,28,33,74,98,120],"AI-based":[4,64,135],"products":[5],"and":[6,41,51,123,134,148,155,189,207,218],"their":[7],"underlying":[8],"models":[9,78],"has":[10,20,145,149],"led":[11],"to":[12,70,143,150,152],"constant":[13],"innovation":[14],"in":[15,37,167],"deep":[16,76,185],"learning":[17,24,77,186],"frameworks.":[18],"Google":[19],"been":[21],"pioneering":[22],"machine":[23],"usage":[25],"across":[26],"dozens":[27],"products.":[29],"Maintaining":[30],"the":[31,49,118,141,162,180,205],"multitude":[32],"model":[34,187],"source":[35],"codes":[36],"different":[38],"ML":[39],"frameworks":[40],"versions":[42],"is":[43],"a":[44,96,168,191],"significant":[45],"challenge.":[46],"So":[47],"far":[48],"maintenance":[50],"migration":[52,73,106],"work":[53],"was":[54],"done":[55],"largely":[56],"manually":[57],"by":[58,117],"human":[59],"experts.":[60],"We":[61,82,202],"describe":[62],"an":[63,91,121],"multi-agent":[65],"system":[66,163],"that":[67,94,113,137,204],"we":[68,88,130,159],"built":[69],"support":[71],"automatic":[72],"TensorFlow-based":[75],"into":[79],"JAX-based":[80],"ones.":[81],"make":[83],"three":[84],"main":[85],"contributions:":[86],"First,":[87],"show":[89],"how":[90,161],"AI":[92,102,196],"planner":[93],"uses":[95],"mix":[97],"static":[99],"analysis":[100],"with":[101],"instructions":[103],"can":[104,211],"create":[105],"plans":[107],"for":[108,184,214],"very":[109],"complex":[110],"code":[111,142,165,220],"components":[112],"are":[114],"reliably":[115],"followed":[116],"combination":[119],"orchestrator":[122],"coders,":[124],"using":[125],"AI-generated":[126],"example-based":[127],"playbooks.":[128],"Second,":[129],"define":[131],"quality":[132],"metrics":[133],"judges":[136],"accelerate":[138],"when":[140],"evaluate":[144],"no":[146],"tests":[147],"adhere":[151],"strict":[153],"style":[154],"dependency":[156],"requirements.":[157],"Third,":[158],"demonstrate":[160],"accelerates":[164],"migrations":[166,188,217],"large":[169],"hyperscaler":[170],"environment":[171],"on":[172],"commercial":[173],"real-world":[174],"use-cases.":[175],"Our":[176],"approach":[177],"dramatically":[178],"reduces":[179],"time":[181],"(6.4x-8x":[182],"speedup)":[183],"creates":[190],"virtuous":[192],"circle":[193],"where":[194],"effectively":[195],"supports":[197],"its":[198],"own":[199],"workflow.":[201],"expect":[203],"techniques":[206],"approaches":[208],"described":[209],"here":[210],"be":[212],"generalized":[213],"other":[215],"framework":[216],"general":[219],"transformation":[221],"tasks.":[222]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-04-02T00:00:00"}