{"id":"https://openalex.org/W7161952545","doi":"https://doi.org/10.48550/arxiv.2605.20577","title":"Mahjax: A GPU-Accelerated Mahjong Simulator for Reinforcement Learning in JAX","display_name":"Mahjax: A GPU-Accelerated Mahjong Simulator for Reinforcement Learning in JAX","publication_year":2026,"publication_date":"2026-05-20","ids":{"openalex":"https://openalex.org/W7161952545","doi":"https://doi.org/10.48550/arxiv.2605.20577"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.20577","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.20577","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":"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.2605.20577","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5091587258","display_name":"Soichiro Nishimori","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Nishimori, Soichiro","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5024903817","display_name":"Shinri Okano","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Okano, Shinri","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5028213867","display_name":"Keigo Habara","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Habara, Keigo","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5061725644","display_name":"Sotetsu Koyamada","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Koyamada, Sotetsu","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5136660150","display_name":"Eason Yu","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Yu, Eason","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5136696221","display_name":"Masashi Sugiyama","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Sugiyama, Masashi","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/T11574","display_name":"Artificial Intelligence in Games","score":0.5329999923706055,"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/T11574","display_name":"Artificial Intelligence in Games","score":0.5329999923706055,"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/T10462","display_name":"Reinforcement Learning in Robotics","score":0.2565000057220459,"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/T11704","display_name":"Mobile Crowdsensing and Crowdsourcing","score":0.012299999594688416,"subfield":{"id":"https://openalex.org/subfields/1706","display_name":"Computer Science Applications"},"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/reinforcement-learning","display_name":"Reinforcement learning","score":0.7470999956130981},{"id":"https://openalex.org/keywords/debugging","display_name":"Debugging","score":0.7440000176429749},{"id":"https://openalex.org/keywords/visualization","display_name":"Visualization","score":0.5805000066757202},{"id":"https://openalex.org/keywords/graphics","display_name":"Graphics","score":0.4652999937534332},{"id":"https://openalex.org/keywords/state","display_name":"State (computer science)","score":0.4408999979496002},{"id":"https://openalex.org/keywords/reinforcement","display_name":"Reinforcement","score":0.4156999886035919},{"id":"https://openalex.org/keywords/rank","display_name":"Rank (graph theory)","score":0.37369999289512634}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7906000018119812},{"id":"https://openalex.org/C97541855","wikidata":"https://www.wikidata.org/wiki/Q830687","display_name":"Reinforcement learning","level":2,"score":0.7470999956130981},{"id":"https://openalex.org/C168065819","wikidata":"https://www.wikidata.org/wiki/Q845566","display_name":"Debugging","level":2,"score":0.7440000176429749},{"id":"https://openalex.org/C36464697","wikidata":"https://www.wikidata.org/wiki/Q451553","display_name":"Visualization","level":2,"score":0.5805000066757202},{"id":"https://openalex.org/C21442007","wikidata":"https://www.wikidata.org/wiki/Q1027879","display_name":"Graphics","level":2,"score":0.4652999937534332},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.4505000114440918},{"id":"https://openalex.org/C48103436","wikidata":"https://www.wikidata.org/wiki/Q599031","display_name":"State (computer science)","level":2,"score":0.4408999979496002},{"id":"https://openalex.org/C107457646","wikidata":"https://www.wikidata.org/wiki/Q207434","display_name":"Human\u2013computer interaction","level":1,"score":0.4320000112056732},{"id":"https://openalex.org/C67203356","wikidata":"https://www.wikidata.org/wiki/Q1321905","display_name":"Reinforcement","level":2,"score":0.4156999886035919},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.3831999897956848},{"id":"https://openalex.org/C164226766","wikidata":"https://www.wikidata.org/wiki/Q7293202","display_name":"Rank (graph theory)","level":2,"score":0.37369999289512634},{"id":"https://openalex.org/C44154836","wikidata":"https://www.wikidata.org/wiki/Q45045","display_name":"Simulation","level":1,"score":0.35899999737739563},{"id":"https://openalex.org/C77660652","wikidata":"https://www.wikidata.org/wiki/Q150971","display_name":"Computer graphics","level":2,"score":0.3517000079154968},{"id":"https://openalex.org/C12725497","wikidata":"https://www.wikidata.org/wiki/Q810247","display_name":"Baseline (sea)","level":2,"score":0.3127000033855438},{"id":"https://openalex.org/C2777211547","wikidata":"https://www.wikidata.org/wiki/Q17141490","display_name":"Training (meteorology)","level":2,"score":0.3009999990463257},{"id":"https://openalex.org/C172367668","wikidata":"https://www.wikidata.org/wiki/Q6504956","display_name":"Data visualization","level":3,"score":0.29840001463890076},{"id":"https://openalex.org/C175154964","wikidata":"https://www.wikidata.org/wiki/Q380077","display_name":"Task analysis","level":3,"score":0.29789999127388},{"id":"https://openalex.org/C2780451532","wikidata":"https://www.wikidata.org/wiki/Q759676","display_name":"Task (project management)","level":2,"score":0.2921000123023987},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.25099998712539673}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.20577","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.20577","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":"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.2605.20577","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.20577","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":"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":[{"id":"https://metadata.un.org/sdg/16","display_name":"Peace, Justice and strong institutions","score":0.8376750946044922}],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Riichi":[0,78],"Mahjong":[1,79],"is":[2],"a":[3,17,75,97],"multi-player,":[4],"imperfect-information":[5],"game":[6],"characterized":[7],"by":[8,64,148],"stochasticity":[9],"and":[10,104,121,135],"high-dimensional":[11],"state":[12],"spaces.":[13],"These":[14],"attributes":[15],"present":[16],"unique":[18],"combination":[19],"of":[20,50,116],"challenges":[21],"that":[22,112,150],"mirror":[23],"complex":[24],"real-world":[25],"decision-making":[26],"problems":[27],"in":[28,82],"reinforcement":[29,146],"learning.":[30],"While":[31],"prior":[32],"research":[33],"has":[34],"heavily":[35],"relied":[36],"on":[37,89,127],"supervised":[38],"learning":[39,51,147],"from":[40],"human":[41],"play":[42],"logs":[43],"to":[44,84,101,118,156],"pre-train":[45],"the":[46,65,133,142],"policy,":[47],"algorithms":[48],"capable":[49],"\\textit{tabula":[52],"rasa}":[53],"(from":[54],"scratch)":[55],"offer":[56],"greater":[57],"potential":[58],"for":[59,145],"general":[60],"applicability,":[61],"as":[62],"evidenced":[63],"AlphaZero":[66],"lineage.":[67],"To":[68],"facilitate":[69],"such":[70],"research,":[71],"we":[72,140],"introduce":[73],"\\textbf{Mahjax},":[74],"fully":[76],"vectorized":[77],"environment":[80],"implemented":[81],"JAX":[83],"enable":[85],"large-scale":[86],"rollout":[87],"parallelization":[88],"Graphics":[90],"Processing":[91],"Units":[92],"(GPUs).":[93],"We":[94],"also":[95],"provide":[96],"high-quality":[98],"visualization":[99],"tool":[100],"streamline":[102],"debugging":[103],"interaction":[105],"with":[106],"trained":[107,154],"agents.":[108],"Experimental":[109],"results":[110],"demonstrate":[111],"Mahjax":[113],"achieves":[114],"throughputs":[115],"up":[117],"\\textbf{2":[119],"million}":[120],"\\textbf{1":[122],"million":[123],"steps":[124],"per":[125],"second}":[126],"eight":[128],"NVIDIA":[129],"A100":[130],"GPUs":[131],"under":[132],"no-red":[134],"red":[136],"rules,":[137],"respectively.":[138],"Furthermore,":[139],"validate":[141],"environment's":[143],"utility":[144],"showing":[149],"agents":[151],"can":[152],"be":[153],"effectively":[155],"improve":[157],"their":[158],"rank":[159],"against":[160],"baseline":[161],"policies.":[162]},"counts_by_year":[],"updated_date":"2026-07-28T07:46:37.118299","created_date":"2026-05-22T00:00:00"}
