JOSS Papers: jax

All Papers
3849
 Published Papers
3274
 Active Papers
575
published Published 7 months ago

jaxKAN: A unified JAX framework for Kolmogorov-Arnold Networks

Python
@srigas
10.21105/joss.07830
published Published 7 months ago

Dynamax: A Python package for probabilistic state space modeling with JAX

Python
@slinderman
10.21105/joss.07069
published Published 8 months ago

unxt: A Python package for unit-aware computing with JAX

Python
@nstarman
10.21105/joss.07771
published Published 8 months ago

PhotonWeave

Python
@SimonSekavcnik
10.21105/joss.07468
published Published 9 months ago

jax-smfsb: A Python library for stochastic systems biology modelling and inference

Python
@darrenjw
10.21105/joss.07491
published Published 10 months ago

SBIAX: Density-estimation simulation-based inference in JAX

Python Roff
@homerjed
10.21105/joss.07606
published Published 11 months ago

DiffOpt: Parallel optimization of Jax models

Python
@AlanPearl
10.21105/joss.07522
published Published about 1 year ago

ReLax: Efficient and Scalable Recourse Explanation Benchmarking using JAX

Python Jupyter Notebook SCSS
@BirkhoffG
10.21105/joss.06567
published Published about 1 year ago

squishyplanet: modeling transits of non-spherical exoplanets in JAX

Python Jupyter Notebook
@ben-cassese
10.21105/joss.06972
published Published over 1 year ago

IKPLS: Improved Kernel Partial Least Squares and Fast Cross-Validation Algorithms for Python with CPU and GPU Implementations Using NumPy and JAX

Python
@Sm00thix
10.21105/joss.06533
Displaying papers 11 - 20 of 36 in total

Journal of Open Source Software is an affiliate of the Open Source Initiative.

Journal of Open Source Software is part of Open Journals, which is a NumFOCUS-sponsored project.

Table of Contents
Public user content licensed CC BY 4.0 unless otherwise specified.
ISSN 2475-9066