{"id":"https://openalex.org/W7134823687","doi":"https://doi.org/10.48550/arxiv.2603.08146","title":"Training event-based neural networks with exact gradients via Differentiable ODE Solving in JAX","display_name":"Training event-based neural networks with exact gradients via Differentiable ODE Solving in JAX","publication_year":2026,"publication_date":"2026-03-09","ids":{"openalex":"https://openalex.org/W7134823687","doi":"https://doi.org/10.48550/arxiv.2603.08146"},"language":null,"primary_location":{"id":"pmh:doi:10.48550/arxiv.2603.08146","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","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":"repository"},"license":"publisher-specific-oa","license_id":"https://openalex.org/licenses/publisher-specific-oa","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":null,"any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5128686361","display_name":"Lukas K\u00f6nig","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"K\u00f6nig, Lukas","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5044645292","display_name":"Manuel Kuhn","orcid":"https://orcid.org/0000-0003-2210-9130"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kuhn, Manuel","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5052732444","display_name":"David Kappel","orcid":"https://orcid.org/0000-0001-5942-4033"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Kappel, David","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5016175782","display_name":"Anand Subramoney","orcid":"https://orcid.org/0000-0002-7333-9860"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Subramoney, Anand","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/T10502","display_name":"Advanced Memory and Neural Computing","score":0.6359999775886536,"subfield":{"id":"https://openalex.org/subfields/2208","display_name":"Electrical and Electronic Engineering"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T10502","display_name":"Advanced Memory and Neural Computing","score":0.6359999775886536,"subfield":{"id":"https://openalex.org/subfields/2208","display_name":"Electrical and Electronic Engineering"},"field":{"id":"https://openalex.org/fields/22","display_name":"Engineering"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10581","display_name":"Neural dynamics and brain function","score":0.14249999821186066,"subfield":{"id":"https://openalex.org/subfields/2805","display_name":"Cognitive Neuroscience"},"field":{"id":"https://openalex.org/fields/28","display_name":"Neuroscience"},"domain":{"id":"https://openalex.org/domains/1","display_name":"Life Sciences"}},{"id":"https://openalex.org/T12611","display_name":"Neural Networks and Reservoir Computing","score":0.131400004029274,"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/ode","display_name":"Ode","score":0.6255999803543091},{"id":"https://openalex.org/keywords/artificial-neural-network","display_name":"Artificial neural network","score":0.6169000267982483},{"id":"https://openalex.org/keywords/biological-neuron-model","display_name":"Biological neuron model","score":0.6011999845504761},{"id":"https://openalex.org/keywords/spike","display_name":"Spike (software development)","score":0.539900004863739},{"id":"https://openalex.org/keywords/differentiable-function","display_name":"Differentiable function","score":0.45590001344680786},{"id":"https://openalex.org/keywords/simple","display_name":"Simple (philosophy)","score":0.4526999890804291},{"id":"https://openalex.org/keywords/quadratic-equation","display_name":"Quadratic equation","score":0.42730000615119934},{"id":"https://openalex.org/keywords/inference","display_name":"Inference","score":0.41600000858306885}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.6855000257492065},{"id":"https://openalex.org/C34862557","wikidata":"https://www.wikidata.org/wiki/Q178985","display_name":"Ode","level":2,"score":0.6255999803543091},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.6169000267982483},{"id":"https://openalex.org/C186565885","wikidata":"https://www.wikidata.org/wiki/Q1651163","display_name":"Biological neuron model","level":3,"score":0.6011999845504761},{"id":"https://openalex.org/C2781390188","wikidata":"https://www.wikidata.org/wiki/Q25203449","display_name":"Spike (software development)","level":2,"score":0.539900004863739},{"id":"https://openalex.org/C202615002","wikidata":"https://www.wikidata.org/wiki/Q783507","display_name":"Differentiable function","level":2,"score":0.45590001344680786},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.4528000056743622},{"id":"https://openalex.org/C2780586882","wikidata":"https://www.wikidata.org/wiki/Q7520643","display_name":"Simple (philosophy)","level":2,"score":0.4526999890804291},{"id":"https://openalex.org/C129844170","wikidata":"https://www.wikidata.org/wiki/Q41299","display_name":"Quadratic equation","level":2,"score":0.42730000615119934},{"id":"https://openalex.org/C2776214188","wikidata":"https://www.wikidata.org/wiki/Q408386","display_name":"Inference","level":2,"score":0.41600000858306885},{"id":"https://openalex.org/C204323151","wikidata":"https://www.wikidata.org/wiki/Q905424","display_name":"Range (aeronautics)","level":2,"score":0.4101000130176544},{"id":"https://openalex.org/C2778794669","wikidata":"https://www.wikidata.org/wiki/Q43054","display_name":"Neuron","level":2,"score":0.3903000056743622},{"id":"https://openalex.org/C2776760102","wikidata":"https://www.wikidata.org/wiki/Q5139990","display_name":"Code (set theory)","level":3,"score":0.32910001277923584},{"id":"https://openalex.org/C28826006","wikidata":"https://www.wikidata.org/wiki/Q33521","display_name":"Applied mathematics","level":1,"score":0.3181999921798706},{"id":"https://openalex.org/C118403218","wikidata":"https://www.wikidata.org/wiki/Q43283","display_name":"Biological neural network","level":2,"score":0.31450000405311584},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.31299999356269836},{"id":"https://openalex.org/C2779227376","wikidata":"https://www.wikidata.org/wiki/Q6505497","display_name":"Layer (electronics)","level":2,"score":0.296999990940094},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.2921999990940094},{"id":"https://openalex.org/C151376022","wikidata":"https://www.wikidata.org/wiki/Q168698","display_name":"Exponential function","level":2,"score":0.2806999981403351},{"id":"https://openalex.org/C11731999","wikidata":"https://www.wikidata.org/wiki/Q9067355","display_name":"Spiking neural network","level":3,"score":0.27810001373291016},{"id":"https://openalex.org/C2777904410","wikidata":"https://www.wikidata.org/wiki/Q7397","display_name":"Software","level":2,"score":0.25110000371932983},{"id":"https://openalex.org/C88548561","wikidata":"https://www.wikidata.org/wiki/Q347599","display_name":"sort","level":2,"score":0.2502000033855438}],"mesh":[],"locations_count":2,"locations":[{"id":"pmh:doi:10.48550/arxiv.2603.08146","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","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":"repository"},"license":"publisher-specific-oa","license_id":"https://openalex.org/licenses/publisher-specific-oa","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},{"id":"doi:10.48550/arxiv.2603.08146","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2603.08146","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":null,"license_id":null,"version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"pmh:doi:10.48550/arxiv.2603.08146","is_oa":true,"landing_page_url":null,"pdf_url":null,"source":{"id":"https://openalex.org/S4406922384","display_name":"Open MIND","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":"repository"},"license":"publisher-specific-oa","license_id":"https://openalex.org/licenses/publisher-specific-oa","version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Article"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Existing":[0],"frameworks":[1],"for":[2,39,97,187,205,229],"gradient-based":[3],"training":[4],"of":[5,132,202,219],"spiking":[6],"neural":[7],"networks":[8],"face":[9],"a":[10,108,129,212,217],"trade-off:":[11],"discrete-time":[12],"methods":[13,31],"using":[14,155],"surrogate":[15],"gradients":[16,35,87],"support":[17],"arbitrary":[18],"neuron":[19,49,99,117,133,157,208,214],"models":[20,158],"but":[21],"introduce":[22,59],"gradient":[23],"bias":[24],"and":[25,42,55,121,137,153,171,180,189],"constrain":[26],"spike-time":[27],"resolution,":[28],"while":[29],"continuous-time":[30],"that":[32,88,215],"compute":[33,86],"exact":[34,90,195],"require":[36],"analytical":[37],"expressions":[38],"spike":[40,75,119],"times":[41],"state":[43],"evolution,":[44],"restricting":[45],"them":[46],"to":[47,85,93],"simple":[48,109],"types":[50,209],"such":[51,159],"as":[52,160],"Leaky":[53,161],"Integrate":[54],"Fire":[56],"(LIF).":[57],"We":[58,145,197],"the":[60,94,116,200],"Eventax":[61,124,147],"framework,":[62],"which":[63,140],"resolves":[64],"this":[65,203],"trade-off":[66],"by":[67,102,210],"combining":[68],"differentiable":[69],"numerical":[70],"ODE":[71],"solvers":[72],"with":[73,91,177,194],"event-based":[74,191],"handling.":[76],"Built":[77],"in":[78,222],"JAX,":[79],"our":[80],"frame-work":[81],"uses":[82,216],"Diffrax":[83],"ODE-solvers":[84],"are":[89],"respect":[92],"forward":[95],"simulation":[96],"any":[98],"model":[100,218],"defined":[101],"ODEs":[103],".":[104],"It":[105],"also":[106,198],"provides":[107],"API":[110],"where":[111],"users":[112],"can":[113,141],"specify":[114],"just":[115],"dynamics,":[118],"conditions,":[120],"reset":[122],"rules.":[123],"prioritises":[125],"modelling":[126],"flexibility,":[127],"supporting":[128],"wide":[130],"range":[131],"models,":[134],"loss":[135,182],"functions,":[136,183],"network":[138],"architectures,":[139],"be":[142],"easily":[143],"extended.":[144],"demonstrate":[146,199],"on":[148],"multiple":[149],"benchmarks,":[150],"including":[151],"Yin-Yang":[152],"MNIST,":[154],"diverse":[156],"Integrate-and-fire":[162,165],"(LIF),":[163],"Quadratic":[164],"(QIF),":[166],"Exponential":[167],"integrate-and-fire":[168],"(EIF),":[169],"Izhikevich":[170],"Event-based":[172],"Gated":[173],"Recurrent":[174],"Unit":[175],"(EGRU)":[176],"both":[178],"time-to-first-spike":[179],"state-based":[181],"demonstrating":[184],"its":[185],"utility":[186],"prototyping":[188],"testing":[190],"architectures":[192],"trained":[193],"gradients.":[196],"application":[201],"framework":[204],"more":[206],"complex":[207],"implementing":[211],"multi-compartment":[213],"dendritic":[220],"spikes":[221],"human":[223],"layer":[224],"2/3":[225],"cortical":[226],"Pyramidal":[227],"neurons":[228],"computation.":[230],"Code":[231],"available":[232],"at":[233],"https://github.com/efficient-scalable-machine-learning/eventax.":[234]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-03-11T00:00:00"}