{"id":"https://openalex.org/W7133334504","doi":"https://doi.org/10.21105/joss.09441","title":"JAX-bandflux: differentiable supernovae SALT modelling for cosmological analysis on GPUs","display_name":"JAX-bandflux: differentiable supernovae SALT modelling for cosmological analysis on GPUs","publication_year":2026,"publication_date":"2026-03-03","ids":{"openalex":"https://openalex.org/W7133334504","doi":"https://doi.org/10.21105/joss.09441"},"language":null,"primary_location":{"id":"doi:10.21105/joss.09441","is_oa":true,"landing_page_url":"https://doi.org/10.21105/joss.09441","pdf_url":"https://joss.theoj.org/papers/10.21105/joss.09441.pdf","source":{"id":"https://openalex.org/S4210214273","display_name":"The Journal of Open Source Software","issn_l":"2475-9066","issn":["2475-9066"],"is_oa":true,"is_in_doaj":true,"is_core":true,"host_organization":"https://openalex.org/P4310315853","host_organization_name":"Open Journals","host_organization_lineage":["https://openalex.org/P4310315853"],"host_organization_lineage_names":["Open Journals"],"type":"journal"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Journal of Open Source Software","raw_type":"journal-article"},"type":"article","indexed_in":["crossref","doaj"],"open_access":{"is_oa":true,"oa_status":"diamond","oa_url":"https://joss.theoj.org/papers/10.21105/joss.09441.pdf","any_repository_has_fulltext":false},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5119848565","display_name":"Samuel Alan Kossoff Leeney","orcid":null},"institutions":[{"id":"https://openalex.org/I241749","display_name":"University of Cambridge","ror":"https://ror.org/013meh722","country_code":"GB","type":"education","lineage":["https://openalex.org/I241749"]}],"countries":["GB"],"is_corresponding":true,"raw_author_name":"Samuel Alan Kossoff Leeney","raw_affiliation_strings":["Astrophysics Group, Cavendish Laboratory, J. J. Thomson Avenue, Cambridge CB3 0HE, UK","Kavli Institute for Cosmology, Madingley Road, Cambridge CB3 0HA, UK"],"raw_orcid":"https://orcid.org/0000-0003-4366-1119","affiliations":[{"raw_affiliation_string":"Astrophysics Group, Cavendish Laboratory, J. J. Thomson Avenue, Cambridge CB3 0HE, UK","institution_ids":["https://openalex.org/I241749"]},{"raw_affiliation_string":"Kavli Institute for Cosmology, Madingley Road, Cambridge CB3 0HA, UK","institution_ids":["https://openalex.org/I241749"]}]}],"institutions":[],"countries_distinct_count":1,"institutions_distinct_count":1,"corresponding_author_ids":["https://openalex.org/A5119848565"],"corresponding_institution_ids":["https://openalex.org/I241749"],"apc_list":null,"apc_paid":null,"fwci":7.1343,"has_fulltext":true,"cited_by_count":1,"citation_normalized_percentile":{"value":0.95782047,"is_in_top_1_percent":false,"is_in_top_10_percent":true},"cited_by_percentile_year":{"min":95,"max":97},"biblio":{"volume":"11","issue":"119","first_page":"9441","last_page":"9441"},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T11323","display_name":"Gamma-ray bursts and supernovae","score":0.703499972820282,"subfield":{"id":"https://openalex.org/subfields/3103","display_name":"Astronomy and Astrophysics"},"field":{"id":"https://openalex.org/fields/31","display_name":"Physics and Astronomy"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T11323","display_name":"Gamma-ray bursts and supernovae","score":0.703499972820282,"subfield":{"id":"https://openalex.org/subfields/3103","display_name":"Astronomy and Astrophysics"},"field":{"id":"https://openalex.org/fields/31","display_name":"Physics and Astronomy"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10026","display_name":"Galaxies: Formation, Evolution, Phenomena","score":0.09790000319480896,"subfield":{"id":"https://openalex.org/subfields/3103","display_name":"Astronomy and Astrophysics"},"field":{"id":"https://openalex.org/fields/31","display_name":"Physics and Astronomy"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T12917","display_name":"Astronomy and Astrophysical Research","score":0.05400000140070915,"subfield":{"id":"https://openalex.org/subfields/3105","display_name":"Instrumentation"},"field":{"id":"https://openalex.org/fields/31","display_name":"Physics and Astronomy"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/supernova","display_name":"Supernova","score":0.6015999913215637},{"id":"https://openalex.org/keywords/differentiable-function","display_name":"Differentiable function","score":0.5985999703407288},{"id":"https://openalex.org/keywords/cosmology","display_name":"Cosmology","score":0.3571000099182129},{"id":"https://openalex.org/keywords/r-package","display_name":"R package","score":0.3425000011920929},{"id":"https://openalex.org/keywords/cosmological-model","display_name":"Cosmological model","score":0.30570000410079956}],"concepts":[{"id":"https://openalex.org/C121332964","wikidata":"https://www.wikidata.org/wiki/Q413","display_name":"Physics","level":0,"score":0.6933000087738037},{"id":"https://openalex.org/C127592171","wikidata":"https://www.wikidata.org/wiki/Q3937","display_name":"Supernova","level":2,"score":0.6015999913215637},{"id":"https://openalex.org/C202615002","wikidata":"https://www.wikidata.org/wiki/Q783507","display_name":"Differentiable function","level":2,"score":0.5985999703407288},{"id":"https://openalex.org/C121864883","wikidata":"https://www.wikidata.org/wiki/Q677916","display_name":"Statistical physics","level":1,"score":0.4036000072956085},{"id":"https://openalex.org/C26405456","wikidata":"https://www.wikidata.org/wiki/Q338","display_name":"Cosmology","level":2,"score":0.3571000099182129},{"id":"https://openalex.org/C2984074130","wikidata":"https://www.wikidata.org/wiki/Q73539779","display_name":"R package","level":2,"score":0.3425000011920929},{"id":"https://openalex.org/C33332235","wikidata":"https://www.wikidata.org/wiki/Q18362","display_name":"Theoretical physics","level":1,"score":0.321399986743927},{"id":"https://openalex.org/C2992891135","wikidata":"https://www.wikidata.org/wiki/Q18346","display_name":"Cosmological model","level":3,"score":0.30570000410079956},{"id":"https://openalex.org/C3020440742","wikidata":"https://www.wikidata.org/wiki/Q1176855","display_name":"Software package","level":3,"score":0.290800005197525},{"id":"https://openalex.org/C44870925","wikidata":"https://www.wikidata.org/wiki/Q37547","display_name":"Astrophysics","level":1,"score":0.2881999909877777},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.2703999876976013},{"id":"https://openalex.org/C2776371256","wikidata":"https://www.wikidata.org/wiki/Q12370","display_name":"Salt (chemistry)","level":2,"score":0.2623000144958496},{"id":"https://openalex.org/C37914503","wikidata":"https://www.wikidata.org/wiki/Q156495","display_name":"Mathematical physics","level":1,"score":0.2515999972820282},{"id":"https://openalex.org/C74650414","wikidata":"https://www.wikidata.org/wiki/Q11397","display_name":"Classical mechanics","level":1,"score":0.25}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.21105/joss.09441","is_oa":true,"landing_page_url":"https://doi.org/10.21105/joss.09441","pdf_url":"https://joss.theoj.org/papers/10.21105/joss.09441.pdf","source":{"id":"https://openalex.org/S4210214273","display_name":"The Journal of Open Source Software","issn_l":"2475-9066","issn":["2475-9066"],"is_oa":true,"is_in_doaj":true,"is_core":true,"host_organization":"https://openalex.org/P4310315853","host_organization_name":"Open Journals","host_organization_lineage":["https://openalex.org/P4310315853"],"host_organization_lineage_names":["Open Journals"],"type":"journal"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Journal of Open Source Software","raw_type":"journal-article"}],"best_oa_location":{"id":"doi:10.21105/joss.09441","is_oa":true,"landing_page_url":"https://doi.org/10.21105/joss.09441","pdf_url":"https://joss.theoj.org/papers/10.21105/joss.09441.pdf","source":{"id":"https://openalex.org/S4210214273","display_name":"The Journal of Open Source Software","issn_l":"2475-9066","issn":["2475-9066"],"is_oa":true,"is_in_doaj":true,"is_core":true,"host_organization":"https://openalex.org/P4310315853","host_organization_name":"Open Journals","host_organization_lineage":["https://openalex.org/P4310315853"],"host_organization_lineage_names":["Open Journals"],"type":"journal"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Journal of Open Source Software","raw_type":"journal-article"},"sustainable_development_goals":[{"id":"https://metadata.un.org/sdg/13","score":0.6697241067886353,"display_name":"Climate action"}],"awards":[],"funders":[],"has_content":{"grobid_xml":true,"pdf":true},"content_urls":{"pdf":"https://content.openalex.org/works/W7133334504.pdf","grobid_xml":"https://content.openalex.org/works/W7133334504.grobid-xml"},"referenced_works_count":3,"referenced_works":["https://openalex.org/W3152905009","https://openalex.org/W4307626599","https://openalex.org/W4415318323"],"related_works":[],"abstract_inverted_index":{"JAX-bandflux":[0],"is":[1,78],"a":[2,31],"JAX":[3],"(Bradbury":[4],"et":[5,27,59,62,104],"al.,":[6,28,60,63,105],"2018)":[7],"implementation":[8],"of":[9,21,48,56,110],"critical":[10],"supernova":[11,49],"modelling":[12],"functionality":[13],"for":[14,67],"cosmological":[15,68],"analysis.The":[16],"codebase":[17],"implements":[18],"key":[19],"components":[20],"the":[22,54,83],"established":[23],"library":[24],"SNCosmo":[25],"(Barbary":[26],"2016)":[29],"in":[30,73,80],"differentiable":[32,46],"framework,":[33],"offering":[34],"efficient":[35],"parallelisation":[36],"and":[37],"gradient-based":[38],"optimisation":[39],"capabilities":[40],"through":[41],"GPU":[42,76],"acceleration.The":[43],"package":[44],"facilitates":[45],"computation":[47],"light":[50],"curve":[51],"measurements,":[52],"supporting":[53],"inference":[55],"SALT":[57],"(Kenworthy":[58],"2021;Pierel":[61],"2022)":[64],"parameters":[65,70],"necessary":[66],"analysis.When":[69],"are":[71],"evaluated":[72],"batches":[74],"on":[75],"(as":[77],"standard":[79],"JAX-based":[81],"samplers),":[82],"bandflux":[84],"kernels":[85],"deliver":[86],"order-of-magnitude":[87],"speedups":[88],"(":[89],"100":[90],"per":[91],"parameter":[92],"set":[93],"compared":[94],"with":[95],"SNCosmo)":[96],"while":[97],"matching":[98],"fluxes":[99],"to":[100],"within":[101],"0.001%":[102],"(Leeney":[103],"2025),":[106],"enabling":[107],"fast":[108],"evaluation":[109],"large,":[111],"fused":[112],"likelihoods.":[113]},"counts_by_year":[{"year":2026,"cited_by_count":1}],"updated_date":"2026-07-02T09:51:11.867554","created_date":"2026-02-02T00:00:00"}
