{"id":"https://openalex.org/W4367365775","doi":"https://doi.org/10.48550/arxiv.2304.14082","title":"JaxPruner: A concise library for sparsity research","display_name":"JaxPruner: A concise library for sparsity research","publication_year":2023,"publication_date":"2023-04-27","ids":{"openalex":"https://openalex.org/W4367365775","doi":"https://doi.org/10.48550/arxiv.2304.14082"},"language":"en","primary_location":{"id":"pmh:oai:arXiv.org:2304.14082","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2304.14082","pdf_url":"https://arxiv.org/pdf/2304.14082","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":"","raw_type":"text"},"type":"preprint","indexed_in":["arxiv","datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://arxiv.org/pdf/2304.14082","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5114048041","display_name":"Joo Hyung Lee","orcid":null},"institutions":[],"countries":[],"is_corresponding":true,"raw_author_name":"Lee, Joo Hyung","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5013461388","display_name":"Wonpyo Park","orcid":"https://orcid.org/0000-0003-0675-6362"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Park, Wonpyo","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5103262879","display_name":"Nicole Mitchell","orcid":"https://orcid.org/0000-0002-6667-2024"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Mitchell, Nicole","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5049628803","display_name":"Jonathan Pilault","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Pilault, Jonathan","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5054834689","display_name":"Johan Obando-Ceron","orcid":"https://orcid.org/0000-0002-6608-5401"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Obando-Ceron, Johan","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5072445293","display_name":"Hanbyul Kim","orcid":"https://orcid.org/0000-0002-8931-4561"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kim, Han-Byul","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5065666573","display_name":"Namhoon Lee","orcid":"https://orcid.org/0000-0001-8999-5603"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Lee, Namhoon","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5060660774","display_name":"Elias Frantar","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Frantar, Elias","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":null,"display_name":"Long, Yun","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Long, Yun","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5070172290","display_name":"Amir Yazdanbakhsh","orcid":"https://orcid.org/0000-0001-8199-7671"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yazdanbakhsh, Amir","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5033596864","display_name":"Shivani Agrawal","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Agrawal, Shivani","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5103255393","display_name":"Suvinay Subramanian","orcid":"https://orcid.org/0000-0002-8715-8964"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Subramanian, Suvinay","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5100327844","display_name":"Xin Wang","orcid":"https://orcid.org/0000-0001-8639-3818"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Wang, Xin","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5035130804","display_name":"Sheng-Chun Kao","orcid":"https://orcid.org/0000-0001-7928-9027"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kao, Sheng-Chun","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5010481801","display_name":"Xingyao Zhang","orcid":"https://orcid.org/0000-0002-8874-9520"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhang, Xingyao","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5032764769","display_name":"Trevor Gale","orcid":"https://orcid.org/0000-0003-3927-9267"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Gale, Trevor","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5071824041","display_name":"Aart J. C. Bik","orcid":"https://orcid.org/0000-0002-0333-7413"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Bik, Aart","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5083809775","display_name":"Woohyun Han","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Han, Woohyun","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5051038221","display_name":"Milen Ferev","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Ferev, Milen","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5043621779","display_name":"Zhonglin Han","orcid":"https://orcid.org/0000-0001-8795-7953"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Han, Zhonglin","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5081959378","display_name":"Hong\u2010Seok Kim","orcid":"https://orcid.org/0000-0002-2356-8403"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kim, Hong-Seok","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5020785659","display_name":"Yann Dauphin","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Dauphin, Yann","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":null,"display_name":"Dziugaite, Gintare Karolina","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Dziugaite, Gintare Karolina","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5068291173","display_name":"Pablo Samuel Castro","orcid":"https://orcid.org/0000-0002-3206-336X"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Castro, Pablo Samuel","raw_affiliation_strings":[],"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5069716783","display_name":"Utku Evci","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Evci, Utku","raw_affiliation_strings":[],"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":25,"corresponding_author_ids":["https://openalex.org/A5114048041"],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":true,"cited_by_count":1,"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/T12072","display_name":"Machine Learning and Algorithms","score":0.9919000267982483,"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"}},"topics":[{"id":"https://openalex.org/T12072","display_name":"Machine Learning and Algorithms","score":0.9919000267982483,"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/T12535","display_name":"Machine Learning and Data Classification","score":0.989799976348877,"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/T10036","display_name":"Advanced Neural Network Applications","score":0.9824000000953674,"subfield":{"id":"https://openalex.org/subfields/1707","display_name":"Computer Vision and Pattern Recognition"},"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/computer-science","display_name":"Computer science","score":0.855791449546814},{"id":"https://openalex.org/keywords/pruning","display_name":"Pruning","score":0.6268748641014099},{"id":"https://openalex.org/keywords/implementation","display_name":"Implementation","score":0.6185887455940247},{"id":"https://openalex.org/keywords/latency","display_name":"Latency (audio)","score":0.5280787944793701},{"id":"https://openalex.org/keywords/overhead","display_name":"Overhead (engineering)","score":0.5198574662208557},{"id":"https://openalex.org/keywords/machine-learning","display_name":"Machine learning","score":0.5112476348876953},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.4376906454563141},{"id":"https://openalex.org/keywords/programming-language","display_name":"Programming language","score":0.294547975063324}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.855791449546814},{"id":"https://openalex.org/C108010975","wikidata":"https://www.wikidata.org/wiki/Q500094","display_name":"Pruning","level":2,"score":0.6268748641014099},{"id":"https://openalex.org/C26713055","wikidata":"https://www.wikidata.org/wiki/Q245962","display_name":"Implementation","level":2,"score":0.6185887455940247},{"id":"https://openalex.org/C82876162","wikidata":"https://www.wikidata.org/wiki/Q17096504","display_name":"Latency (audio)","level":2,"score":0.5280787944793701},{"id":"https://openalex.org/C2779960059","wikidata":"https://www.wikidata.org/wiki/Q7113681","display_name":"Overhead (engineering)","level":2,"score":0.5198574662208557},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.5112476348876953},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.4376906454563141},{"id":"https://openalex.org/C199360897","wikidata":"https://www.wikidata.org/wiki/Q9143","display_name":"Programming language","level":1,"score":0.294547975063324},{"id":"https://openalex.org/C6557445","wikidata":"https://www.wikidata.org/wiki/Q173113","display_name":"Agronomy","level":1,"score":0.0},{"id":"https://openalex.org/C86803240","wikidata":"https://www.wikidata.org/wiki/Q420","display_name":"Biology","level":0,"score":0.0},{"id":"https://openalex.org/C76155785","wikidata":"https://www.wikidata.org/wiki/Q418","display_name":"Telecommunications","level":1,"score":0.0}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:oai:arXiv.org:2304.14082","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2304.14082","pdf_url":"https://arxiv.org/pdf/2304.14082","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":"","raw_type":"text"},{"id":"doi:10.48550/arxiv.2304.14082","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2304.14082","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":"article"}],"best_oa_location":{"id":"pmh:oai:arXiv.org:2304.14082","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2304.14082","pdf_url":"https://arxiv.org/pdf/2304.14082","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":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":"","raw_type":"text"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"grobid_xml":true,"pdf":true},"content_urls":{"pdf":"https://content.openalex.org/works/W4367365775.pdf","grobid_xml":"https://content.openalex.org/works/W4367365775.grobid-xml"},"referenced_works_count":0,"referenced_works":[],"related_works":["https://openalex.org/W2120447654","https://openalex.org/W2977179488","https://openalex.org/W2144453115","https://openalex.org/W2128223750","https://openalex.org/W4238532390","https://openalex.org/W2188872161","https://openalex.org/W2961779879","https://openalex.org/W797688974","https://openalex.org/W2002978035","https://openalex.org/W2209382646"],"abstract_inverted_index":{"This":[0],"paper":[1],"introduces":[2],"JaxPruner,":[3],"an":[4],"open-source":[5],"JAX-based":[6],"pruning":[7,31],"and":[8,32,39,50,86,88],"sparse":[9,22,33],"training":[10,34],"library":[11,57],"for":[12],"machine":[13],"learning":[14],"research.":[15],"JaxPruner":[16,45],"aims":[17],"to":[18],"accelerate":[19],"research":[20],"on":[21,92],"neural":[23],"networks":[24],"by":[25,76],"providing":[26,77],"concise":[27],"implementations":[28],"of":[29,74],"popular":[30,55,93],"algorithms":[35],"with":[36,53,65],"minimal":[37],"memory":[38],"latency":[40],"overhead.":[41],"Algorithms":[42],"implemented":[43],"in":[44,60,79],"use":[46],"a":[47],"common":[48],"API":[49],"work":[51],"seamlessly":[52],"the":[54],"optimization":[56],"Optax,":[58],"which,":[59],"turn,":[61],"enables":[62],"easy":[63],"integration":[64,75],"existing":[66],"JAX":[67],"based":[68],"libraries.":[69],"We":[70],"demonstrate":[71],"this":[72],"ease":[73],"examples":[78],"four":[80],"different":[81],"codebases:":[82],"Scenic,":[83],"t5x,":[84],"Dopamine":[85],"FedJAX":[87],"provide":[89],"baseline":[90],"experiments":[91],"benchmarks.":[94]},"counts_by_year":[{"year":2025,"cited_by_count":1}],"updated_date":"2026-04-17T18:11:37.981687","created_date":"2025-10-10T00:00:00"}