JOSS Papers: jax

All Papers
3936
 Published Papers
3306
 Active Papers
630
published Published 4 days ago

Bijx: Bijections and normalizing flows with JAX/NNX

Python
@mathisgerdes
10.21105/joss.09521
published Published 26 days ago

jaxhps: An elliptic PDE solver built with machine learning in mind

Python
@meliao
10.21105/joss.08549
published Published about 1 month ago

MDPax: GPU-accelerated MDP solvers in Python with JAX

Python
@joefarrington
10.21105/joss.08897
published Published about 1 month ago

Bamojax: Bayesian Modelling with JAX

Python Jupyter Notebook
@maxhinne
10.21105/joss.08642
published Published about 1 month ago

TMMax: High-performance modeling of multilayer thin-film structures using transfer matrix method with JAX

Python Jupyter Notebook
@bahremsd
10.21105/joss.09088
published Published about 2 months ago

JaxLayerLumos: A JAX-based Differentiable Optical and Radio Frequency Simulator for Multilayer Structures

Python
@Mil152
10.21105/joss.08572
published Published 3 months ago

Graphem - JAX: Node Influence Maximization via Geometric Embeddings

Python
@sashakolpakov
10.21105/joss.08855
published Published 4 months ago

ginjax: E(d)-Equivariant CNN for Tensor Images

Jupyter Notebook Python
@WilsonGregory
10.21105/joss.08129
published Published 4 months ago

smolyax: a high-performance implementation of the Smolyak interpolation operator in JAX

Jupyter Notebook Python
@JoWestermann
10.21105/joss.08505
published Published 6 months ago

DiRe - JAX: A JAX based Dimensionality Reduction Algorithm for Large-scale Data

Python
@sashakolpakov
10.21105/joss.08264
Displaying papers 1 - 10 of 38 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