{"id":"https://openalex.org/W4291127115","doi":"https://doi.org/10.1145/3534678.3539065","title":"Rax: Composable Learning-to-Rank Using JAX","display_name":"Rax: Composable Learning-to-Rank Using JAX","publication_year":2022,"publication_date":"2022-08-12","ids":{"openalex":"https://openalex.org/W4291127115","doi":"https://doi.org/10.1145/3534678.3539065"},"language":"en","primary_location":{"id":"doi:10.1145/3534678.3539065","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3534678.3539065","pdf_url":null,"source":{"id":"https://openalex.org/S4363608767","display_name":"Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"conference"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining","raw_type":"proceedings-article"},"type":"article","indexed_in":["crossref"],"open_access":{"is_oa":true,"oa_status":"hybrid","oa_url":"https://doi.org/10.1145/3534678.3539065","any_repository_has_fulltext":false},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5042795557","display_name":"Rolf Jagerman","orcid":"https://orcid.org/0000-0002-5169-495X"},"institutions":[{"id":"https://openalex.org/I1291425158","display_name":"Google (United States)","ror":"https://ror.org/00njsd438","country_code":"US","type":"company","lineage":["https://openalex.org/I1291425158","https://openalex.org/I4210128969"]}],"countries":["US"],"is_corresponding":true,"raw_author_name":"Rolf Jagerman","raw_affiliation_strings":["Google Research, Mountain View, CA, USA"],"affiliations":[{"raw_affiliation_string":"Google Research, Mountain View, CA, USA","institution_ids":["https://openalex.org/I1291425158"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5064608039","display_name":"Xuanhui Wang","orcid":"https://orcid.org/0009-0000-1388-1423"},"institutions":[{"id":"https://openalex.org/I1291425158","display_name":"Google (United States)","ror":"https://ror.org/00njsd438","country_code":"US","type":"company","lineage":["https://openalex.org/I1291425158","https://openalex.org/I4210128969"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Xuanhui Wang","raw_affiliation_strings":["Google Research, Mountain View, CA, USA"],"affiliations":[{"raw_affiliation_string":"Google Research, Mountain View, CA, USA","institution_ids":["https://openalex.org/I1291425158"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5011279860","display_name":"Honglei Zhuang","orcid":"https://orcid.org/0000-0001-8134-1509"},"institutions":[{"id":"https://openalex.org/I1291425158","display_name":"Google (United States)","ror":"https://ror.org/00njsd438","country_code":"US","type":"company","lineage":["https://openalex.org/I1291425158","https://openalex.org/I4210128969"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Honglei Zhuang","raw_affiliation_strings":["Google Research, Mountain View, CA, USA"],"affiliations":[{"raw_affiliation_string":"Google Research, Mountain View, CA, USA","institution_ids":["https://openalex.org/I1291425158"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5100763095","display_name":"Zhen Qin","orcid":"https://orcid.org/0000-0001-7857-9719"},"institutions":[{"id":"https://openalex.org/I1291425158","display_name":"Google (United States)","ror":"https://ror.org/00njsd438","country_code":"US","type":"company","lineage":["https://openalex.org/I1291425158","https://openalex.org/I4210128969"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Zhen Qin","raw_affiliation_strings":["Google Research, Mountain View, CA, USA"],"affiliations":[{"raw_affiliation_string":"Google Research, Mountain View, CA, USA","institution_ids":["https://openalex.org/I1291425158"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5032248436","display_name":"Michael Bendersky","orcid":"https://orcid.org/0000-0002-2941-6240"},"institutions":[{"id":"https://openalex.org/I1291425158","display_name":"Google (United States)","ror":"https://ror.org/00njsd438","country_code":"US","type":"company","lineage":["https://openalex.org/I1291425158","https://openalex.org/I4210128969"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Michael Bendersky","raw_affiliation_strings":["Google Research, Mountain View, CA, USA"],"affiliations":[{"raw_affiliation_string":"Google Research, Mountain View, CA, USA","institution_ids":["https://openalex.org/I1291425158"]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5037200145","display_name":"Marc Najork","orcid":"https://orcid.org/0000-0003-1423-0854"},"institutions":[{"id":"https://openalex.org/I1291425158","display_name":"Google (United States)","ror":"https://ror.org/00njsd438","country_code":"US","type":"company","lineage":["https://openalex.org/I1291425158","https://openalex.org/I4210128969"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Marc Najork","raw_affiliation_strings":["Google Research, Mountain View, CA, USA"],"affiliations":[{"raw_affiliation_string":"Google Research, Mountain View, CA, USA","institution_ids":["https://openalex.org/I1291425158"]}]}],"institutions":[],"countries_distinct_count":1,"institutions_distinct_count":6,"corresponding_author_ids":["https://openalex.org/A5042795557"],"corresponding_institution_ids":["https://openalex.org/I1291425158"],"apc_list":null,"apc_paid":null,"fwci":0.7276,"has_fulltext":false,"cited_by_count":8,"citation_normalized_percentile":{"value":0.70986856,"is_in_top_1_percent":false,"is_in_top_10_percent":false},"cited_by_percentile_year":{"min":89,"max":99},"biblio":{"volume":null,"issue":null,"first_page":"3051","last_page":"3060"},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T10028","display_name":"Topic Modeling","score":0.9998999834060669,"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/T10028","display_name":"Topic Modeling","score":0.9998999834060669,"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/T12072","display_name":"Machine Learning and Algorithms","score":0.9991000294685364,"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/T11307","display_name":"Domain Adaptation and Few-Shot Learning","score":0.9972000122070312,"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/ranking","display_name":"Ranking (information retrieval)","score":0.8208717107772827},{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.6746280789375305},{"id":"https://openalex.org/keywords/rank","display_name":"Rank (graph theory)","score":0.6081891655921936},{"id":"https://openalex.org/keywords/set","display_name":"Set (abstract data type)","score":0.5334686636924744},{"id":"https://openalex.org/keywords/learning-to-rank","display_name":"Learning to rank","score":0.4784679114818573},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.462862104177475},{"id":"https://openalex.org/keywords/flexibility","display_name":"Flexibility (engineering)","score":0.44631442427635193},{"id":"https://openalex.org/keywords/function","display_name":"Function (biology)","score":0.43009382486343384},{"id":"https://openalex.org/keywords/machine-learning","display_name":"Machine learning","score":0.4180137515068054},{"id":"https://openalex.org/keywords/programming-language","display_name":"Programming language","score":0.17512235045433044},{"id":"https://openalex.org/keywords/statistics","display_name":"Statistics","score":0.13107511401176453},{"id":"https://openalex.org/keywords/biology","display_name":"Biology","score":0.12753784656524658},{"id":"https://openalex.org/keywords/mathematics","display_name":"Mathematics","score":0.11496180295944214}],"concepts":[{"id":"https://openalex.org/C189430467","wikidata":"https://www.wikidata.org/wiki/Q7293293","display_name":"Ranking (information retrieval)","level":2,"score":0.8208717107772827},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6746280789375305},{"id":"https://openalex.org/C164226766","wikidata":"https://www.wikidata.org/wiki/Q7293202","display_name":"Rank (graph theory)","level":2,"score":0.6081891655921936},{"id":"https://openalex.org/C177264268","wikidata":"https://www.wikidata.org/wiki/Q1514741","display_name":"Set (abstract data type)","level":2,"score":0.5334686636924744},{"id":"https://openalex.org/C86037889","wikidata":"https://www.wikidata.org/wiki/Q4330127","display_name":"Learning to rank","level":3,"score":0.4784679114818573},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.462862104177475},{"id":"https://openalex.org/C2780598303","wikidata":"https://www.wikidata.org/wiki/Q65921492","display_name":"Flexibility (engineering)","level":2,"score":0.44631442427635193},{"id":"https://openalex.org/C14036430","wikidata":"https://www.wikidata.org/wiki/Q3736076","display_name":"Function (biology)","level":2,"score":0.43009382486343384},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.4180137515068054},{"id":"https://openalex.org/C199360897","wikidata":"https://www.wikidata.org/wiki/Q9143","display_name":"Programming language","level":1,"score":0.17512235045433044},{"id":"https://openalex.org/C105795698","wikidata":"https://www.wikidata.org/wiki/Q12483","display_name":"Statistics","level":1,"score":0.13107511401176453},{"id":"https://openalex.org/C86803240","wikidata":"https://www.wikidata.org/wiki/Q420","display_name":"Biology","level":0,"score":0.12753784656524658},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.11496180295944214},{"id":"https://openalex.org/C114614502","wikidata":"https://www.wikidata.org/wiki/Q76592","display_name":"Combinatorics","level":1,"score":0.0},{"id":"https://openalex.org/C78458016","wikidata":"https://www.wikidata.org/wiki/Q840400","display_name":"Evolutionary biology","level":1,"score":0.0}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.1145/3534678.3539065","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3534678.3539065","pdf_url":null,"source":{"id":"https://openalex.org/S4363608767","display_name":"Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"conference"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining","raw_type":"proceedings-article"}],"best_oa_location":{"id":"doi:10.1145/3534678.3539065","is_oa":true,"landing_page_url":"https://doi.org/10.1145/3534678.3539065","pdf_url":null,"source":{"id":"https://openalex.org/S4363608767","display_name":"Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining","issn_l":null,"issn":null,"is_oa":false,"is_in_doaj":false,"is_core":false,"host_organization":null,"host_organization_name":null,"host_organization_lineage":[],"host_organization_lineage_names":[],"type":"conference"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining","raw_type":"proceedings-article"},"sustainable_development_goals":[{"display_name":"Life in Land","id":"https://metadata.un.org/sdg/15","score":0.5099999904632568}],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":28,"referenced_works":["https://openalex.org/W1985757471","https://openalex.org/W2004881074","https://openalex.org/W2047221353","https://openalex.org/W2059001985","https://openalex.org/W2069870183","https://openalex.org/W2091158010","https://openalex.org/W2094145178","https://openalex.org/W2108862644","https://openalex.org/W2143331230","https://openalex.org/W2146502635","https://openalex.org/W2295598076","https://openalex.org/W2566147423","https://openalex.org/W2902365885","https://openalex.org/W2955421345","https://openalex.org/W2981852735","https://openalex.org/W2994069334","https://openalex.org/W3023202929","https://openalex.org/W3028556448","https://openalex.org/W3094600186","https://openalex.org/W3100107515","https://openalex.org/W3101267375","https://openalex.org/W3101537642","https://openalex.org/W3105136066","https://openalex.org/W3137439872","https://openalex.org/W4205254860","https://openalex.org/W4288089799","https://openalex.org/W6639094827","https://openalex.org/W6785566885"],"related_works":["https://openalex.org/W3127142483","https://openalex.org/W2963493716","https://openalex.org/W2572749905","https://openalex.org/W4297665600","https://openalex.org/W2293317945","https://openalex.org/W4323349240","https://openalex.org/W1786507113","https://openalex.org/W4318960487","https://openalex.org/W2499321295","https://openalex.org/W2275578992"],"abstract_inverted_index":{"Rax":[0,15,32,54,73,126,137,175],"is":[1,16],"a":[2,34,56,76,100,155],"library":[3,176],"for":[4,109],"composable":[5],"Learning-to-Rank":[6],"(LTR)":[7],"written":[8],"entirely":[9],"in":[10,117],"JAX.":[11,31],"The":[12],"goal":[13],"of":[14,21,30,37,49,58,66,84,125,184],"to":[17,103],"facilitate":[18],"easy":[19],"prototyping":[20],"LTR":[22,141],"systems":[23],"by":[24,127],"leveraging":[25],"the":[26,47,50,123,165,174,178,185],"flexibility":[27],"and":[28,41,69,91,134,144],"simplicity":[29],"provides":[33,74,99],"diverse":[35],"set":[36],"popular":[38,140],"ranking":[39,67,86,107,110,151,162,169],"metrics":[40,87,111],"losses":[42,68,152],"that":[43,80,94,112,149],"integrate":[44],"well":[45],"with":[46,153,177],"rest":[48],"JAX":[51,187],"ecosystem.":[52],"Furthermore,":[53,146],"implements":[55],"system":[57],"ranking-specific":[59],"function":[60,77],"transformations":[61],"which":[62],"allows":[63],"fine-grained":[64],"customization":[65],"metrics.":[70],"Most":[71],"notably":[72],"approx_t12n:":[75],"transformation":[78],"(t12n)":[79],"can":[81,95,159],"transform":[82],"any":[83],"our":[85],"into":[88],"an":[89],"approximate":[90],"differentiable":[92],"form":[93],"be":[96],"optimized.":[97],"This":[98],"systematic":[101],"way":[102],"directly":[104],"optimize":[105],"neural":[106,129],"models":[108,130],"are":[113,172],"not":[114],"easily":[115],"optimizable":[116],"other":[118],"libraries.":[119],"We":[120,171],"empirically":[121],"demonstrate":[122],"effectiveness":[124],"benchmarking":[128],"implemented":[131],"using":[132,136],"Flax":[133],"trained":[135],"on":[138,164],"two":[139],"benchmarks:":[142],"WEB30K":[143],"Istella.":[145],"we":[147],"show":[148],"integrating":[150],"T5,":[154],"large":[156],"language":[157],"model,":[158],"improve":[160],"overall":[161],"performance":[163],"MS":[166],"MARCO":[167],"passage":[168],"task.":[170],"sharing":[173],"open":[179],"source":[180],"community":[181],"as":[182],"part":[183],"larger":[186],"ecosystem":[188],"at":[189],"https://github.com/google/rax.":[190]},"counts_by_year":[{"year":2026,"cited_by_count":1},{"year":2025,"cited_by_count":3},{"year":2024,"cited_by_count":3},{"year":2023,"cited_by_count":1}],"updated_date":"2025-11-06T03:46:38.306776","created_date":"2025-10-10T00:00:00"}